leetcode算法(199.二叉树的右视图)

层序遍历的时候,判断是否遍历到单层的最后面的元素,如果是,就放进result数组中,随后返回result就可以了。

cpp 复制代码
class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        // 创建一个队列用于BFS层序遍历
        queue<TreeNode*> que;
        
        // 如果根节点不为空,将其加入队列
        if (root != NULL) que.push(root);
        
        // 存储结果的向量,将保存二叉树的右视图
        vector<int> result;
        
        // 当队列不为空时,继续遍历
        while (!que.empty()) {
            // 获取当前层的节点数量
            int size = que.size();
            
            // 遍历当前层的所有节点
            for (int i = 0; i < size; i++) {
                // 取出队首节点
                TreeNode* node = que.front();
                que.pop();
                
                // 如果是当前层的最后一个节点(最右侧节点),将其值加入结果
                // 因为队列是先进先出,所以最后一个节点就是该层最右边的节点
                if (i == (size - 1)) result.push_back(node->val);
                
                // 将当前节点的左子节点加入队列(如果存在)
                if (node->left) que.push(node->left);
                
                // 将当前节点的右子节点加入队列(如果存在)
                if (node->right) que.push(node->right);
            }
        }
        
        // 返回右视图结果
        return result;
    }
};
相关推荐
IronMurphy15 小时前
【算法三十九】994. 腐烂的橘子
算法
Ares-Wang16 小时前
算法》》旅行商问题 TSP、7座桥问题 哈密顿回路 深度优先 和 宽度优先
算法·深度优先·宽度优先
Liqiuyue16 小时前
Transformer:现代AI革命背后的核心模型
人工智能·算法·机器学习
WolfGang00732116 小时前
代码随想录算法训练营 Day34 | 动态规划 part07
算法·动态规划
Kk.080217 小时前
Linux(十一)fork实例练习、文件操作示例及相关面试题目分享
linux·运维·算法
潇冉沐晴18 小时前
2026CCCC第三次模拟赛 部分题解
算法
WolfGang00732118 小时前
代码随想录算法训练营 Day32 | 动态规划 part05
算法·动态规划
碧海银沙音频科技研究院18 小时前
1-1杰理蓝牙SOC的UI配置开发方法
人工智能·深度学习·算法
啊我不会诶19 小时前
2024CCPC长春邀请赛
算法