问题背景
给定一个二叉树的 根节点 r o o t root root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
数据约束
- 二叉树的节点个数的范围是 [ 0 , 100 ] [0,100] [0,100]
- − 100 ≤ N o d e . v a l ≤ 100 -100 \le Node.val \le 100 −100≤Node.val≤100
解题过程
右视图当然考虑三种深度优先遍历的做法,稍微修改一下先序遍历就可以实现。
如果没做过做这个题,记录结果判断条件可能不一定好想,当成模板记下来。多维护一个深度,当深度和结果集长度不匹配时,说明是第一次到达某一层,这时候把当前节点加入到结果集。
具体实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<>();
dfs(root, 0, res);
return res;
}
private void dfs(TreeNode root, int depth, List<Integer> res) {
if(root == null) {
return;
}
if(depth == res.size()) {
res.add(root.val);
}
dfs(root.right, depth + 1, res);
dfs(root.left, depth + 1, res);
}
}