199. 二叉树的右视图

https://leetcode.cn/problems/binary-tree-right-side-view/description/?envType=study-plan-v2&envId=top-interview-150

其实就是让我们返回每一层最右边的节点的,首先想到的就是把每一层的节点从左往右排列出来然后添加最后一个节点,直到把所有层添加完。怎么获取各个层的节点呢,我们可以使用二叉树的层序遍历,使用队列来存储每一层的节点

class Solution {
    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;
        }
    }

    public List<Integer> rightSideView(TreeNode root) {
        if(root == null) return new ArrayList<>();
        if(root.left == null && root.right == null) return List.of(root.val);
        Queue<TreeNode> queue1 = new LinkedList<>();//存储当前层的节点
        Queue<TreeNode> queue2 = new LinkedList<>();//存储下一层的节点
        queue1.add(root);
        List<Integer> res = new ArrayList<>();
        while (!queue1.isEmpty()) {
            while(!queue1.isEmpty()){
                if(queue1.size() == 1) { // 如果当前层只有一个节点,那么这个节点就是最右边的节点
                    res.add(queue1.peek().val);
                }
                TreeNode cur = queue1.poll();
                // 填充下一层
                if(cur.left != null) queue2.add(cur.left);
                if(cur.right != null) queue2.add(cur.right);
            }
            //遍历完当前层后,交换queue1和queue2,遍历下一层
            Queue<TreeNode> temp = queue1;
            queue1 = queue2;
            queue2 = temp;
        }
        return res;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值