LC 116/117填充每个节点的下一个右侧节点指针（层序遍历的应用）

题目：

给定一个二叉树

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

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

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

 

进阶：

• 你只能使用常量级额外空间。
• 使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

 

示例：

输入：root = [1,2,3,4,5,null,7]
输出：[1,#,2,3,#,4,5,7,#]
解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。

 

提示：

• 树中的节点数小于 6000
• -100 <= node.val <= 100

 

思路：

看到这个题首先想到的是：就是按照层序遍历的方式，将每一层的元素取出来之后链接起来就ok了

那么如何应用层序遍历方法来实现这个功能呢？

 

我们来看一下原始的层序遍

while(!q_data.empty())
{
    Node* temp = q_data.front();// 取出队首元素
    q_data.pop();    // 删除队首元素
    // 进行操作
    .......

    // 将当前队首元素的子节点添加到队列中
    if(temp->left) q_data.push(temp->left);
    if(temp->right) q_data.push(temp->right);
    ...
    over
}

对于这两道题来说，我们需要在上面这个框架的基础下，增加将每一层元素取出，并连接的过程；

这就涉及到一个问题：如何确定是同一层的两个节点

1. 可以使用类似pair <> 的节点和层数对儿，这样相对浪费空间，有木有更好的？

2. 在每个while中，我们先计算当前q_data的大小size，代表当前层数节点个数，之后在for循环中取出前面size个元素

（注意：因为在每次for循环中我们都要向其中添加新的子节点，因此，不能使用q_data是否为空来判断是不是到了这一层的最后一个节点，应该使用for循环中的i和size-1比较；

下面看代码：

代码：

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL)    return NULL;
        
        queue<Node*> q_data;
        q_data.push(root);    // 先把头节点添加进去
        while(!q_data.empty())
        {
            int size = q_data.size();    // 计算当前queue中有几个元素
            for(int i = 0; i < size; i++){    // 对于每一个元素先取出，再根据是否为当前层的最后一个元素，链接到下一个元素上或是NULL
                Node* temp = q_data.front();
                q_data.pop();
                if(i == size - 1) temp->next = NULL;
                else {
                    temp->next = q_data.front();
                }
                // 别忘了在每个for里面需要将pop节点的子节点添加到队列中
                if(temp->left != NULL) q_data.push(temp->left);
                if(temp->right != NULL) q_data.push(temp->right);
            }         
        }
        return root;
    }
};