 LeetCode199:二叉树的右视图

LeetCode199:二叉树的右视图

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

 

示例 1：

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

输出: [1,3,4]

关键词：二叉树的层序遍历，队列（先进先出）

思路：二叉树的右视图中的元素 等价于  二叉树的层序遍历中每一层的最后一个节点。

基本步骤：准备一个栈，根节点入队，队列不为空的情况下一直执行以下操作：

记录当前队列的大小（即同层节点的个数）。出队，如果当前出队元素是同层中的最后一个节点，就将此元素添加到结果链表res中。循环遍历队列中的TreeNode节点，如果节点存在左右子树，就将其左右子树加入到队列中。

代码实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List rightSideView(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        Queue queue = new LinkedList<>();
        List res = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            for (int i = 0; i < size; ++i) {
                TreeNode cur = queue.poll();
                if (i == size - 1) {
                    res.add(cur.val);
                }
                if (cur.left != null) {
                    queue.offer(cur.left);
                }
                if (cur.right != null) {
                    queue.offer(cur.right);
                }
            }
        }
        return res;
    }
}