【力扣每日一题】116. 填充每个节点的下一个右侧节点指针
原题链接
思路:层序遍历,每次扩展当前层 全部节点
class Solution {
public:
Node* connect(Node* root) {
queue<Node *>Q;
if(root==NULL)return root;
Q.push(root);
while(!Q.empty())
{
int size = Q.size();
for(int i = 0; i < size; i++) { //核心步骤
Node* node = Q.front();
Q.pop();
if (i < size - 1) { //链接next指针
node->next = Q.front();
}
if (node->left != nullptr) {
Q.push(node->left);
}
if (node->right != nullptr) {
Q.push(node->right);
}
}
}
return root;
}
};