LeetCode199

问题

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

思路

相当于二叉树的按层级遍历的进阶，我们按照之前文章写过的二叉树层级遍历：https://blog.youkuaiyun.com/C_lea/article/details/152413818?spm=1001.2014.3001.5501
只要我们在每一层循环的时候，判断一下当前循环的是不是最后一个节点即可，只要是最后一个节点，那么就把结果加入进去

for (int i = 0;i < n;i ++) {
    TreeNode t = queue.poll();
    //...
    if (i == n - 1) {
        result.add(t.val);   //如果是当前层的最后一个节点，就相当于是右视图，那么加入结果
    }
}

代码

public List<Integer> rightSideView(TreeNode root) {
    List<Integer> result = new ArrayList<>();
    Queue<TreeNode> queue = new LinkedList<>();
    if (root == null) {
        return result;
    }
    queue.offer(root);
    while (!queue.isEmpty()) {
        int n = queue.size();
        for (int i = 0;i < n;i ++) {
            TreeNode t = queue.poll();
            if (t.left != null) {
                queue.offer(t.left);
            }
            if (t.right != null) {
                queue.offer(t.right);
            }
            if (i == n - 1) {
                result.add(t.val);
            }
        }
    }
    return result;
}