leetcode算法(116.填充每个节点的下一个右侧节点指针)

cpp 复制代码
 for(int i=0 ; i<size ; i++){
     Node *node = que.front();
     que.pop();
     Node *node2 = que.front();
// 尝试获取队列中的下一个节点(下一层的第一个节点或当前层的下一个节点)
// 注意:这里有一个逻辑错误❌ ,当i=size-1时,que.front()可能不存在
// 因为当前层的最后一个节点已经出队,队列中可能是下一层的节点

     if(i==size-1){
        node->next=NULL;
     } 
     else{
        node->next=node2;
     } 

改进代码

cpp 复制代码
class Solution {
public:
    Node* connect(Node* root) {
        // 创建一个队列用于层序遍历
        queue<Node*> que;
        
        // 如果根节点不为空,将其加入队列
        if(root!=NULL) que.push(root);
        
        // 当队列不为空时,继续处理每一层
        while(!que.empty()){
            // 获取当前层的节点数量
            int size=que.size();
            
            // 遍历当前层的所有节点
            for(int i=0 ; i<size ; i++){
                // 从队列中取出当前节点
                Node *node = que.front();
                que.pop();
                
                // 判断是否为当前层的最后一个节点
                if(i==size-1){
                    // 如果是最后一个节点,将其next指针设为NULL
                    node->next=NULL;
                } 
                else{
                    // 如果不是最后一个节点
                    // 查看队列中下一个节点(即当前层的下一个兄弟节点)
                    // 注意:此时不弹出队列中的下一个节点,因为会在下一次循环中处理
                    Node *node2 = que.front();
                    // 将当前节点的next指针指向下一个兄弟节点
                    node->next=node2;
                } 
                
                // 如果当前节点有左子节点,将其加入队列(为下一层做准备)
                if(node->left) que.push(node->left);
                
                // 如果当前节点有右子节点,将其加入队列(为下一层做准备)
                if(node->right) que.push(node->right);
            }
        }
        // 返回连接好的树的根节点
        return root;
    }
};
相关推荐
UIU1147 小时前
P1233 [ICPC 2001 Taejon R] 木棍加工
数据结构·c++·算法·动态规划
论文复现现场7 小时前
WRF、FEA 能用 L40S 加速吗?求解器检查、断点恢复与长期运行
java·数据库·算法·wrf·有限元分析·gpu计算·断点续算
ZhengEnCi7 小时前
用 AI 写代码为什么拼到最后还是拼人:算力瓶颈与领域专家提示词的本质差异
算法·架构
秋天的怀念_David8 小时前
第二十一夜 · 数字黑洞与西西弗斯的惩罚
java·程序人生·算法
碧海银沙音频科技研究院9 小时前
在PC电脑上用Whisper→Marian算法实现离线语音翻译功能
算法·whisper
niyongsheng9 小时前
3D 装箱算法工程化:数据结构、放置启发式与支撑规则
数据结构·算法·rust
叉烧滑蛋11 小时前
B站台湾大学郭彦甫|MATLAB 学习笔记|04 变量与档案存取
数据结构
j7~12 小时前
【穷举、深搜与回溯算法】一棵决策树讲透全排列与子集
数据结构·c++·算法·leetcode·回溯·递归
傅科摆 _ py13 小时前
DP,贪心,堆和栈,二分力扣复习
开发语言·python·算法
Shaoshing13 小时前
LeetCode100 42.接雨水
java·算法·leetcode