(LeetCode 面试经典 150 题) 117. 填充每个节点的下一个右侧节点指针 II（广度优先搜索bfs）

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

思路：广度优先搜索bfs，时间复杂度0(n)，空间复杂度0(1)。

既然当前行都已经Next了，那么下一行可以通过当前行来Next到一起。

C++版本：

/*
// 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 root;
        Node * p=root;
        while(p){
        	// nd.Next用来记录下一行的首节点
            Node nd(0);
            // 维护下一行的Next
            Node * cur=&nd;
            Node *q=p;
            // 当前行
            while(q){
            	// 左子树不为空
                if(q->left!=nullptr){
                    cur->next=q->left;
                    cur=cur->next;
                }
                // 右子树不为空
                if(q->right!=nullptr){
                    cur->next=q->right;
                    cur=cur->next;
                }
                // 当前行的下一个节点
                q=q->next;
            }
            // 下一行
            p=nd.next;
        }
        return root;
    }
};

JAVA版本：

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

    public Node() {}
    
    public Node(int _val) {
        val = _val;
    }

    public 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 root;
        Node p=root;
        while(p!=null){
            Node nd=new Node(0);
            Node cur=nd;
            Node q=p;
            while(q!=null){
                if(q.left!=null){
                    cur.next=q.left;
                    cur=cur.next;
                }
                if(q.right!=null){
                    cur.next=q.right;
                    cur=cur.next;
                }
                q=q.next;
            }
            p=nd.next;
        }
        return root;
    }
}

GO版本：

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Left *Node
 *     Right *Node
 *     Next *Node
 * }
 */

func connect(root *Node) *Node {
	if root==nil {return root}
    p:=root
    for p!=nil {
        nd:=Node{Val:0,Next:nil}
        cur:=&nd
        for p!=nil {
            if p.Left !=nil {
                cur.Next=p.Left
                cur=cur.Next
            }
            if p.Right !=nil {
                cur.Next=p.Right
                cur=cur.Next
            }
            p=p.Next
        }
        p=nd.Next
    }
    return root
}