题目:给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
本题采用层序遍历,取每一层的最右边元素。
public List<Integer> rightSideView(TreeNode root) {
List<Integer> list = new ArrayList<Integer>();
Queue<TreeNode> queue = new LinkedList<TreeNode>();
if(root==null){
return list;
}
queue.offer(root);
while(!queue.isEmpty()){
int size = queue.size();
for(int i=0;i<size;i++){
TreeNode temp = queue.poll();
if(i==0){
list.add(temp.val);
}
if(temp.right!=null){
queue.offer(temp.right);
}
if(temp.left!=null){
queue.offer(temp.left);
}
}
}
return list;
}