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

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

解题思路

思路同116题一样

代码

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

执行用时：8 ms, 在所有 C++ 提交中击败了95.74%的用户
内存消耗：17.1 MB, 在所有 C++ 提交中击败了41.10%的用户
通过测试用例：55 / 55