[Leetcode] Binary Tree Right Side View 二叉树右视图

Binary Tree Right Side View

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.

For example: Given the following binary tree,

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

You should return [1, 3, 4].

深度优先搜索

复杂度

时间 O(b^(h+1)-1) 空间 O(h) 递归栈空间 对于二叉树b=2

思路

本题实际上是求二叉树每一层的最后一个节点，我们用DFS先遍历右子树并记录遍历的深度，如果这个右子节点的深度大于当前所记录的最大深度，说明它是下一层的最右节点（因为我们先遍历右边，所以每一层都是先从最右边进入），将其加入结果中。

代码

public class Solution {
    
    int maxdepth = 0;
    List<Integer> res;
    
    public List<Integer> rightSideView(TreeNode root) {
        res = new LinkedList<Integer>();
        if(root!=null) helper(root,1);
        return res;
    }
    
    private void helper(TreeNode root, int depth){
        if(depth>maxdepth){
            maxdepth = depth;
            res.add(root.val);
        }
        if(root.right!=null) helper(root.right, depth+1);
        if(root.left!=null) helper(root.left, depth+1);
    }
    
}

层序遍历 Level Order Traversal

复杂度

时间 O(b^(h+1)-1) 空间 O(b^h) 对于二叉树b=2

思路

我们同样可以借用层序遍历的思路，只要每次把这一层的最后一个元素取出来就行了，具体代码参见Binary Tree Traversal中的Binary Tree Level Order Traversal