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

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

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

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

进阶：

你只能使用常量级额外空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

题解：

方法一：采用层次遍历。

/*
// 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 null;
        
        Queue<Node> queue = new LinkedList<Node>();

        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i = 0; i < size; i++){
                Node top1 = queue.poll();
                if(i == size - 1)
                    top1.next = null;
                else
                    top1.next = queue.peek();
                
                if(top1.left != null)
                    queue.offer(top1.left);
                if(top1.right != null)
                    queue.offer(top1.right);
            }
        }
        return root;
    }
}

上述方法的时间复杂度为O(n),用队列保存节点，因此空间复杂度也为O(n).

方法二：可以将空间复杂度优化到O(1)。具体做法：

第一种情况为：连接同一父节点的两个子节点。假设node为父节点，则可以：
	node.left.next = node.right;

第二种情况为：node和node1为相邻节点，node在左，即node.next = node1。这时连接下一层时，需要连接
	node.right.next = node.next.left;

从上往下建立， 即在N-1层为N层建立连接。每次都找到当前层的最左侧节点，从左往右连接，根节点root不需要建立。

/*
// 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 null;
        Node leftmost = root;

        while(leftmost.left != null){
            Node head = leftmost;
            while(head != null){
                //建立第一种连接
                head.left.next = head.right;
                //建立第二种连接
                if(head.next != null)
                    head.right.next = head.next.left;
                
                head = head.next;
            }
            leftmost = leftmost.left;
        }  
      
        return root;
    }
}

题目链接：https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node
题解参考：https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node/solution/tian-chong-mei-ge-jie-dian-de-xia-yi-ge-you-ce-2-4/