【LeetCode 随笔】面试经典 150 题【中等+困难】持续更新中。。。-199.中等二叉树的右视图

时间:2024-07-20 07:18:40

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

示例 1:
在这里插入图片描述
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]


示例 2:
输入: [1,null,3]
输出: [1,3]


示例 3:
输入: []
输出: []

class Solution {  
    public List<Integer> rightSideView(TreeNode root) {  
        List<Integer> ans = new ArrayList<>(); // 初始化一个ArrayList来存储结果  
        if (root == null) return ans; // 如果根节点为空,则直接返回空列表  
          
        Queue<TreeNode> q = new LinkedList<>(); // 创建一个队列用于BFS遍历  
        q.offer(root); // 将根节点加入队列  
          
        while (!q.isEmpty()) { // 当队列不为空时,继续遍历  
            int count = q.size(); // 获取当前层的节点数  
            for (int i = 0; i < count; i++) { // 遍历当前层的所有节点  
                TreeNode node = q.poll(); // 从队列中取出一个节点  
                if (node.left != null) q.offer(node.left); // 如果左子节点不为空,则将其加入队列  
                if (node.right != null) q.offer(node.right); // 如果右子节点不为空,则将其加入队列  
                // 当遍历到当前层的最后一个节点时,将其值添加到结果列表中  
                if (i == count - 1) ans.add(node.val);   
            }  
        }  
        return ans; 
    }  
}