二叉树的右视图
二叉树的层序遍历,添加判断,将每层最后一个找出来就行。
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<>();
if(root == null) return res;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
int size = queue.size();
while(size-->0){
TreeNode node = queue.poll();
if(size == 0){//添加当前层最后一个节点
res.add(node.val);
}
if(node.left != null) queue.offer(node.left);
if(node.right != null) queue.offer(node.right);
}
}
return res;
}
}