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