DEV Community

loading...
Cover image for Solution: Binary Tree Right Side View

Solution: Binary Tree Right Side View

seanpgallivan
Fledgling software developer; the struggle is a Rational Approximation.
・2 min read

This is part of a series of Leetcode solution explanations (index). If you liked this solution or found it useful, please like this post and/or upvote my solution post on Leetcode's forums.


Leetcode Problem #199 (Medium): Binary Tree Right Side View


Description:

Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.


Examples:

Example:
Input: [1,2,3,null,5,null,4]
Output: [1, 3, 4]
Visual: Example Visual

Idea:

A problem dealing with tree traversal typically means a breadth first search or a depth first search approach. Since we're tasked with isolating a value from each level, this would naturally bring to mind the BFS approach... but let's keep DFS on the back burner; we'll come back to it.

A BFS approach usually entails the use of a queue (q) where we push each node's children onto q while we move along a level of the tree from one side to the other. This ensures that when each level is completed, we have the next level ready to go in q. In order to separate each level, since we're continuously adding to q, we can just take the length of q at the beginning of each level to figure out when the next one begins.

In this case, we can run our BFS from right to left and simply push the first value of each level in our answer array (ans) before returning the array upon complete traversal of the tree.

But what about the DFS approach? DFS solutions often allow us to find a concise, recursive solution, and while they're not always the first thought when it comes to tree traversal problems where the level is important, in this case we don't need the level as a whole, we just need one end of each level.

This means that we can define a recursive function (dfs) to run through the nodes from left to right and simply overwrite the value for each level (ans[lvl]) as we get to each node, since the last value from left to right on each level will be the one we want to keep.


Javascript Code w/ BFS:

var rightSideView = function(root) {
    let q = [root], ans = []
    while (q[0]) {
        ans.push(q[0].val)
        for (let i = 0, len = q.length; i < len; i++) {
            let curr = q.shift()
            if (curr.right) q.push(curr.right)
            if (curr.left) q.push(curr.left)
        }
    }
    return ans
};
Enter fullscreen mode Exit fullscreen mode

Javascript Code w/ DFS:

var rightSideView = function(root) {
    let ans = []
    const dfs = (node, lvl) => {
        if (!node) return
        ans[lvl] = node.val
        dfs(node.left, lvl+1)
        dfs(node.right, lvl+1)
    }
    dfs(root, 0)
    return ans
};
Enter fullscreen mode Exit fullscreen mode

Discussion (0)