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

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

问题描述

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

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

层次遍历

与BFS相似，但BFS每次只取队列中的一个元素进行拓展，而层次遍历每次将队列的所有元素都出来拓展，这样保证每次从队列中拿出来的元素都是属于同一层的。next指针初始状态被设置为NULL，故每一层的最后一个节点的next指针不用进行改变。但是我们需要其孩子的信息还是需要遍历。

Node* connect(Node* root){
    if(root==nullptr){
        return root;
    }
    queue<Node*> Q;
    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){
                node->next = Q.front();
            }
            if(node->left!=nullptr){
                Q.push(node->left);
            }
            if(node->right!=nullptr){
                Q.push(node->right);
            }
        }
    }
    return root;
}

复杂度分析

时间复杂度O(N),每个节点被访问一次。

空间复杂度O(N)，取决于一个层级上的最大元素数量。

使用已经建立的next指针

连接第N+1层的next指针可以在第N层进行

两种情况
第N层节点的左孩子的next指针指向右孩子， CONNECTION1
右孩子的next指针指向该节点next指针指向的左孩子，直至next指针为空 CONNECTION2

Node* connect(Node* root){
    if(!root){
        return root;
    }
    //START FROM ROOT
    Node* leftMost = root;
    //FIRST BASE CASE
    while(leftMost->left){
        Node* head = leftMost;
        //SECOND BASE CASE
        while(head){
            //CONNECTION1
            head->left->next = head->right;

            //CONNECTION2
            if(head->next){
                head->right->next = head->next->left;
            }
            head = head->next;
        }
        leftMost = leftMost->left;
    }
    return root;
}

复杂度分析

时间复杂度为O(n)，每个节点只访问一次

空间复杂度为O(1),不需要储存额外的节点