题目链接:https://leetcode-cn.com/problems/binary-tree-right-side-view/
题目描述
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
第一次编辑代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if(root == null)
return ans;
ans.add(root.val);
rlook(root ,ans, 1);
return ans;
}
public void rlook(TreeNode root, List ans, int depth){
if(root != null){
if(depth > ans.size()){
ans.add(root.val);
}
rlook(root.right, ans, depth + 1);
rlook(root.left, ans, depth + 1);
}
}
}
反思
做完发现大家都是用层次遍历。
优化了代码结构。
第二次编辑代码:
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> ans = new ArrayList<>();
rlook(root ,ans, 1);
return ans;
}
public void rlook(TreeNode root, List ans, int depth){
if(root == null)
return;
if(depth > ans.size())
ans.add(root.val);
rlook(root.right, ans, depth + 1);
rlook(root.left, ans, depth + 1);
}
}