给定一个二叉树的 根节点 root
,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
思路:其实就是寻找二叉树每一层的最右边节点。
加一个判断语句即可,判断当前节点是否为最右侧节点。
public static List<Integer> rightSideView(TreeNode root) {
List<Integer> res=new ArrayList();
//借助队列
Queue<TreeNode> queue=new LinkedList();
if(root!=null){
queue.add(root);
}
//思路:判断当前节点是否为每一层的最右边节点
while (!queue.isEmpty()){
int size=queue.size(); //记录每层个数
for(int i=0;i<size;i++){
TreeNode node=queue.poll();
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
if(i==size-1){
res.add(node.val);
}
}
}
return res;
}