题目描述
这个例子比较迷惑。
用下面这个:
那就是用队列整一个层序遍历就完事了。
思路 1
直接上代码吧。
public class Solution {
private final List<Integer> result = new ArrayList<>();
public List<Integer> rightSideView(TreeNode root) {
if (root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
queue.offer(null);
while (!queue.isEmpty()) {
TreeNode poll = queue.poll();
if (poll == null) continue; // 应该没有必要,但是还是加上去防止人为胡乱操作队列引发错误。
if (poll.left != null) {
queue.offer(poll.left);
}
if (poll.right != null) {
queue.offer(poll.right);
}
if (queue.peek() == null) { // 判断是不是到了该层末尾。
queue.poll(); // 处理掉 null 值
result.add(poll.val);
queue.offer(null);
}
}
return result;
}
}