填充完美二叉树的next指针

填充完美二叉树的next指针

思路

• 对于root节点，只要让root.left.next = root.right即可。
• 但要让节点5和节点6相连，必须从各自的父亲入手，因此递归函数需要传入两个参数，设为node1和node2，连接node1和node2需要三步
• 1、node1与node2相连：node1.next = node2
• 2、各自的左右孩子相连：node1.left.next = node1.right，node2.left.next = node2
• 3、node1的右孩子与node2的左孩子相连：node1.right.next = node2.left

递归代码

/**
 * // Definition for a Node.
 * function Node(val, left, right, next) {
 *    this.val = val === undefined ? null : val;
 *    this.left = left === undefined ? null : left;
 *    this.right = right === undefined ? null : right;
 *    this.next = next === undefined ? null : next;
 * };
 */

/**
 * @param {Node} root
 * @return {Node}
 */
var connect = function(root) {
    if(root==null) return null;
    if(root.left!=null){
        connectTwoNode(root.left, root.right);
    }
    return root;
};

// 连接同一层上的node1和node2两个节点
var connectTwoNode= function(node1, node2) {
    if(node1!=null){
        node1.next = node2;
        // 连接同一个父亲的左右节点
        connectTwoNode(node1.left, node1.right);
        if(node2!=null){
            connectTwoNode(node2.left, node2.right); 
            // 连接跨父亲的两个节点
            connectTwoNode(node1.right, node2.left);
        }
    }
}