算法题——二叉树的右视图

1. 本题知识点

二叉树、广度优先遍历

2. 题目描述

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

输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

3. 解题思路

使用层次遍历，将每一层的最后一个结点加入返回结果中。

4. 代码

public class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<>();
        if (root == null) {
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            LinkedList<TreeNode> nextLevel = new LinkedList<>();
            int lastVal = 0;
            while (!queue.isEmpty()) {
                TreeNode node = queue.poll();
                lastVal = node.val;
                if (node.left != null) {
                    nextLevel.add(node.left);
                }
                if (node.right != null) {
                    nextLevel.add(node.right);
                }
            }
            result.add(lastVal);
            queue.addAll(nextLevel);
        }
        return result;
    }
}