leetcode116. 填充每个节点的下一个右侧节点指针，搞清解题关键秒解，附详细代码

leetcode116. 填充每个节点的下一个右侧节点指针

给定一个 完美二叉树 ，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

示例 1：

输入：root = [1,2,3,4,5,6,7]
输出：[1,#,2,3,#,4,5,6,7,#]
解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。序列化的输出按层序遍历排列，同一层节点由 next 指针连接，‘#’ 标志着每一层的结束。

示例 2:
输入：root = []
输出：[]

算法思路

本题的解题关键在于你要能想到层序遍历这一手段，题目填充每个节点的下一个右侧节点指针，实质上就是二叉树按层序遍历的每一层之间串了起来，如果能明白这一点的话，这一点能够秒解。

算法步骤

初始化队列：首先检查根节点是否为空，如果为空则直接返回。
根节点入队：将根节点root加入队列q。
循环条件：当队列q不为空时，执行循环。
获取队列大小：在每次循环开始时，获取队列中元素的数量，即当前层的节点数。
遍历当前层节点：使用一个for循环遍历当前层的所有节点，i从0到size-1。
节点出队：每次循环开始时，将队头元素t出队。
连接相邻节点：如果当前节点不是当前层的最后一个节点（即i < size - 1），则将下一个节点设置为当前节点的next指针。
子节点入队：如果当前节点的左子节点或右子节点不为空，则将它们加入队列，以便在下一轮循环中处理。
循环结束：当当前层的所有节点都被处理后，循环结束。
返回根节点：最后返回修改后的根节点。

具体代码

class Solution {
public:
    Node* connect(Node* root) {
    if(!root) return root;
    queue<Node*> q;
    q.push(root);
    while(!q.empty())
    {
        int size=q.size();
        for(int i=0;i<size;i++)
        {
            Node* t=q.front();
            q.pop();
            if(i<size-1)
            {
                t->next=q.front();
            }
            if(t->left!=NULL) q.push(t->left);
            if(t->right!=NULL) q.push(t->right);
        }
    }
    return root;
    }
};