二叉树的右视图

发表于 更新于 分类于

算法-二叉树的右视图

背景

  • 算法–二叉树的右视图

  • 博主以代码随想录算法公开课进行学习

题目

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

    示例 1：

    **输入：**root = [1,2,3,null,5,null,4]

    输出：[1,3,4]

    示例 2：

    **输入：**root = [1,2,3,4,null,null,null,5]

    输出：[1,3,4,5]

思路

  • 与层序遍历差不多，判断是否遍历到单层的最后一个元素，如果是就加入队列

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
/**
 * 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<Integer> rightSideView(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        if(root == null){
            return list;
        }
        que.offerLast(root);
        while(!que.isEmpty()){
            int levelSize = que.size();
            for(int i = 0;i < levelSize; i++){
                TreeNode poll = que.pollFirst();
                if(poll.left != null){
                    que.addLast(poll.left);
                }
                if(poll.right != null){
                    que.addLast(poll.right);
                }
                if(i == levelSize -1){  //位于同层末尾，加入数组
                    list.add(poll.val);
                }

            }
        }

        return list;
    }
}