Alberta Williams I like to write about JavaScript and web development. My interests include artificial intelligence, games, and math. Learn more about me at roboberta.com.

Getting started with recursion for tree traversal

4 min read 1384


Have you ever encountered a problem you felt could be solved with recursion, except you didn’t know where to start? Or did it seem like you had to hack your way to a solution?

The first part of tackling recursion is understanding when a problem calls for it. Recursion can be used when the problem can be modeled as a recurrence relation. A recurrence relation is a rule for finding future values from previous values. The Fibonacci sequence is an example of a recurrence relation. Recursion can also be used when the data is defined recursively. A filesystem can be defined recursively because each directory is made up of other directories.

The second part is understanding how to implement a recursive function. In this post, I will show you techniques for using recursion to traverse recursive data structures.

Finding items in a tree

A recursive data structure is similar to a tree. In code, this translates to an array of arrays or an object whose keys are other objects. Our case study will be a tree that models the neighborhoods in the city of New York. The root of the tree is New York. It has two children, Manhattan and Brooklyn. And Manhattan has two children, Harlem and Upper East Side.

This is the list representation of our tree:

No Title

No Description

We will implement a function, `includes`, to test if our list contains the specified item. The function will return true if it finds a match, otherwise false.

There are three parts to this function. First, the base case. Our function will be reducing the list at each step until we have a list with no elements. Next, is the case when we are looking at an individual node. A node would be the string ‘Manhattan’. Last, is the case when the element is another list or subtree. The list `[‘Harlem’, ‘Upper East Side’]` is a subtree.

This is the skeleton for these three cases:

No Title

No Description

The `isEmpty` function returns `true` if the list has no elements. If all of the elements in the list have been traversed and no match has been found, the function returns `false`. The `first` function returns the first element in the list. The `isNode` function returns `false` if the element is a list.

In the `else if` you want to test if the current element matches the item you are searching for. If it is, you can return true. If it isn’t, you need to recur on the rest of the list.

This is the updated code:

No Title

No Description

The `rest` function returns the list without the first element. This is how we are reducing the problem so that reach the base case, an empty list. The `else if` block of the conditional statement could have also been written as

No Title

No Description

It does the same job, but more succinctly. I prefer this line of code to the nested `if` statements.

Last, in the `else` block we need to recur on the first element because it is a list and recur on the rest of the list. This is the code for the `else` block:

No Title

No Description

Putting it all together you now have:

No Title

No Description

Removing items from a tree

Next, we will implement a function `remove` that takes a string and a list as input and returns the list with all occurrences of the string removed. In a real tree, you might be interested in removing a node along with all of its children. For simplicity, we will only look at the case for removing an individual item.

Removing an item from a list is similar to finding it’s members, except we need to make sure we are keeping a reference to our list as we recur on its subparts.

The three cases will be the same:

No Title

No Description

Because this function returns a list, our base case will return an empty array. The new list will be built by copying all of the items from the list except the item to be removed.

If we were removing an item from a one-dimensional list using a for loop, the function might look like this:

No Title

No Description

For the recursive implementation, the test goes in the `else if` block. If the current element equals the item, we recur on the rest of the list. This has the effect of removing the item. However, if the current element is not the item, then we have to save that part to concatenate to the rest of the list we are recurring on. When the function reaches the base case, all of the concatenations that were deferred will be added to this list.

No Title

No Description

The `concat` function here joins the two inputs into one list.

In the `else` block we define the case where the current element is a list. We need to recur on that part and recur on the rest of the list. Additionally, both parts will need to be concatenated into one list. This is what we end up with:

No Title

No Description

Exercise

Implement a function, `occur`, that takes a string and a list as input and returns the number of times the string appears in the list. First, set up your three cases. What should you return in your base case? What should you do when you have a node? What should you do when you have a list? Use the previous two examples as a guide.

Conclusion

The techniques used for finding and removing items can be extended to solving many other problems that require tree traversal. Trees can be used to model the moves in a game or for performing a binary search. When implementing a recursive function, keep these points in mind:

  • Define the base case
  • Define the case where the element is a node
  • Define the case where the element is a list
  • In the recursive call, change the arguments so that the function reaches the base case

Another point to consider is that recursion may not always be the most efficient way to solve the problem. That is why you should remember that any problem that can be solved using recursion can also be solved using `for` and `while` loops. You would choose recursion over a loop when the benefits of having a simpler solution outweigh the costs of efficiency.

Finally, the examples shown here are just one way to solve these kinds of problems. Use them as a starting point and read the resources listed below for a deeper understanding.

Further Reading


Plug: LogRocket, a DVR for web apps

LogRocket is a frontend logging tool that lets you replay problems as if they happened in your own browser. Instead of guessing why errors happen, or asking users for screenshots and log dumps, LogRocket lets you replay the session to quickly understand what went wrong. It works perfectly with any app, regardless of framework, and has plugins to log additional context from Redux, Vuex, and @ngrx/store.

In addition to logging Redux actions and state, LogRocket records console logs, JavaScript errors, stacktraces, network requests/responses with headers + bodies, browser metadata, and custom logs. It also instruments the DOM to record the HTML and CSS on the page, recreating pixel-perfect videos of even the most complex single page apps.

Try it for free.

Alberta Williams I like to write about JavaScript and web development. My interests include artificial intelligence, games, and math. Learn more about me at roboberta.com.

Leave a Reply