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

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

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

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

思路

层序遍历

一开始居然没有想到层序遍历，直接层序遍历，可以获得当前层所有节点，连线即可。

代码

/*
// 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 == nullptr) {
            return nullptr;
        }
        queue<Node*> q;
        q.push(root);
        while(!q.empty()) {
            int n = q.size();
            //保存上一个节点
            Node* last = nullptr;
            for(int i = 0; i < n; i++) {
                Node* tmp = q.front();
                q.pop();
                if(tmp -> left != nullptr) {
                    q.push(tmp -> left);
                }
                if(tmp -> right != nullptr) {
                    q.push(tmp -> right);
                }
                //访问的不是第一个节点
                //连接
                if(i != 0) {
                    last -> next = tmp;
                }
                last = tmp;
            }
        }
        return root;
    }
};

next节点

可以在访问第i层时，连接好第i+1层的节点信息，访问根节点时，就直接处理第二层的信息即可，这样在访问每一层时，就可以当作链表来访问以及处理下一层的next连接了。

代码

/*
// 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:
	//负责连接，并处理下一层的起始点
    void handle(Node* &last, Node* &p, Node* &nextStart) {
        //last节点存储的是上一个节点信息
        //last不为空，证明访问的不是层第一个节点
        if(last != nullptr) {
            last -> next = p;
        } 
        //nextStart存储的是下一层的起始点，为空证明访问的是层第一个节点
        if(nextStart == nullptr) {
            nextStart = p;
        }
        //更新last节点
        last = p;
    }
    Node* connect(Node* root) {
        if(root == nullptr) {
            return nullptr;
        }
        Node* tmp = root;
        while(tmp) {
            Node* last = nullptr, *nextStart = nullptr;
            //把当前层当作链表访问
            for(Node* p = tmp; p != nullptr; p = p -> next) {
                //处理下一层
                if(p -> left != nullptr) {
                    handle(last, p -> left, nextStart);
                }
                if(p -> right != nullptr) {
                    handle(last, p -> right, nextStart);
                }
            }
            //到达下一层起始点
            tmp = nextStart;
        }
        return root;
    }
};