

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;
}
};