06_二叉树的右视图

发布时间 2023-11-17 15:17:01作者: 鲍宪立

二叉树的右视图

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

示例 1:

img

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

示例 2:

输入: [1,null,3]
输出: [1,3]

示例 3:

输入: []
输出: []

提示:

  • 二叉树的节点个数的范围是 [0,100]
  • -100 <= Node.val <= 100

【思路】

层次遍历的时候,判断是否遍历到单层的有右边的元素,如果是,则放进result数组中,随后返回result即可。

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        if (root == null) return list;
        que.add(root);
        while (!que.isEmpty()) {
            int levelSize = que.size();
            for(int i = 0; i < levelSize; i++) {
                TreeNode tmp = que.poll();
                if (tmp.left != null)  que.add(tmp.left);
                if (tmp.right != null)  que.add(tmp.right);
                //判断i是否为当前层的最右边的元素,是最右边的元素的话返回至结果列表中
                if (i == levelSize - 1) {
                    list.add(tmp.val);
                }
            }   
        }
        return list;
    }
}