给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <---
/ \
2 3 <---
\ \
5 4 <---
题解:
- 从二叉树根结点开始右子树优先深度遍历,辅助栈记录当前遍历路径,deep记录当前深度
- 若list在当前deep所在的层数已存入第一个结点的值,则后续结点均不可见,也即不存入
class Solution {
//在ret的list中,ret[i] = 第i层看到的第一个结点,若不存在表示当前未看到第一个结点或不存在
List<Integer> ret = new ArrayList<>();
//辅助栈 记录从右到左深度遍历的过程
Stack<TreeNode> s = new Stack<>();
public List<Integer> rightSideView(TreeNode root) {
dfs(root,0);
return ret;
}
void dfs(TreeNode root, int deep){//deep = 当前遍历层数
if(root==null){
return ;
}
s.push(root);
while(!s.empty()){
root = s.pop();
if(ret.size()<=deep){//当前层未看到第一个右侧结点
ret.add(root.val);
}
dfs(root.right,deep+1);//从右往左看
dfs(root.left,deep+1);
}
}
}