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

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

解题思路

借助队列，判断队列中的元素是否为二叉树种每层的最后一个元素，进行next指针修改。
所有指针修改完毕，最后直接返回根节点root。

代码

/*
// 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) {
        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(); //出队

                //判断是否为本层最后一个元素，是则设置next为空
                if(i == size-1) node->next = NULL;
                //否则指向队头元素，即为同层的下一个元素
                else node->next = que.front();

                //左右子树入队
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
        }
        return root;
    }
};

执行用时: 16 ms
内存消耗: 16.6 MB