[Java]LeetCode117. 填充同一层的兄弟节点 II | Populating Next Right Pointers in Each Node II

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/9953121.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

热烈欢迎，请直接点击！！！

进入博主App Store主页，下载使用各个作品！！！

注：博主将坚持每月上线一个新app！！！

Given a binary tree

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

Populate each next pointer to point to its next right node. If there is no next right node, the next pointer should be set to NULL.

Initially, all next pointers are set to NULL.

Note:

• You may only use constant extra space.
• Recursive approach is fine, implicit stack space does not count as extra space for this problem.

Example:

Given the following binary tree,

     1
   /  \
  2    3
 / \    \
4   5    7

After calling your function, the tree should look like:

     1 -> NULL
   /  \
  2 -> 3 -> NULL
 / \    \
4-> 5 -> 7 -> NULL

---

给定一个二叉树

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

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

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

说明:

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

示例:

给定二叉树，

     1
   /  \
  2    3
 / \    \
4   5    7

调用你的函数后，该二叉树变为：

     1 -> NULL
   /  \
  2 -> 3 -> NULL
 / \    \
4-> 5 -> 7 -> NULL

---

0ms

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

    public Node() {}

    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) {
         Node node = root;
         while(node != null){
            Node tempChild = new Node(0);
            Node currentChild = tempChild;
            while(node!=null){
                if(node.left != null) { currentChild.next = node.left; currentChild = currentChild.next;}
                if(node.right != null) { currentChild.next = node.right; currentChild = currentChild.next;}
                node = node.next;
            }
            node = tempChild.next;
        }
        return root;
    }
}

---

1ms

class Solution {
    public Node connect(Node root) {
        connect(root, null);
        return root;
    }
    
    private void connect(Node root, Node nxtRight) {
        if(root == null) {
            return;
        }
        
        root.next = nxtRight;
        Node nxtRightForRight = null;
        Node curr = root.next;
        while(curr != null) {
            if(curr.left != null) {
                nxtRightForRight = curr.left;
                break;
            } else if(curr.right != null) {
                nxtRightForRight = curr.right;
                break;
            }
            curr = curr.next;
        }
        connect(root.right, nxtRightForRight);
        connect(root.left, root.right == null ? nxtRightForRight : root.right);
        return;        
    }
}

---

2ms

class Solution {
    public Node connect(Node root) {
        if (root == null) {
            return root;
        }
        return connectHelper(root,0,new HashMap<Integer,Node>());//map put the last node of every level
    }
    
    private Node connectHelper(Node node,int level,Map<Integer,Node> map) {
        
        if (node == null) {
            return node;
        }
        Node last = null;
        if (map.containsKey(level)) {
            last = map.get(level);            
            last.next = node;
        }
        map.put(level,node);
        Node left = connectHelper(node.left,level+1,map);
        Node right = connectHelper(node.right,level+1,map); 
        
        return node;
    }
}

---

3ms

class Solution {
    public Node connect(Node root) {
        if (root == null) {
            return root;
        }
        Deque<Node> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            Node prev = null,cur = null;
            for (int i = 0; i < size; i++) {
                cur = queue.poll();
                if (prev != null) {
                    prev.next = cur;
                }
                prev = cur;
                
                if (cur.left != null) {
                    queue.offer(cur.left);
                }
                if (cur.right != null) {
                    queue.offer(cur.right);
                }
            }
            cur.next = null;
        }
        return root;
    } 
}

---

4ms

class Solution {
    public Node connect(Node root) {
         if(root==null){
            return null;
        }
       LinkedList<Node> stack=new LinkedList<>();
       LinkedList<Node> temp=new LinkedList<>(); 
       List<List<Node>> list=new ArrayList<List<Node>>();
        List<Node> l=new ArrayList<>();
        stack.addFirst(root);
        while(!stack.isEmpty()){
            Node rp=stack.removeFirst();
            l.add(rp);
            if(rp.left!=null){
                temp.addLast(rp.left);
            }
            if(rp.right!=null){
                temp.addLast(rp.right);
            }
            if(stack.isEmpty()){
                stack=temp;
                temp=new LinkedList<>();
                list.add(l);
                l=new ArrayList<>();
            }
        }
        for(int i=0;i<list.size();i++){
            for(int j=0;j<list.get(i).size();j++){
                if(j==list.get(i).size()-1){
                    Node s=list.get(i).get(j);
                    s.next=null;
                    break;
                }
                Node a=list.get(i).get(j);
                a.next=list.get(i).get(j+1);
            }
        }
        return root;
    }
}

---

5ms

class Solution {
    public Node connect(Node root) {
        if (root == null) {
            return null;
        }
        
        Node lastHead = root;
        Node curHead = lastHead.left != null ? lastHead.left : lastHead.right;
        
        Node last = lastHead;
        Node cur = curHead;
        
        while (cur != null) {
            Node nxt = null;
            while (last != null && nxt == null) {
                if (cur == last.left) {
                    if (last.right != null) {
                        nxt = last.right;
                    } else {
                        last = last.next;
                    }
                } else if (cur == last.right) {
                    last = last.next;
                } else {
                    if (last.left != null) {
                        nxt = last.left;
                    } else if (last.right != null) {
                        nxt = last.right;
                    } else {
                        last = last.next;
                    }
                }
            }
            
            cur.next = nxt;
            cur = cur.next;
            
            // System.out.println(cur != null ? cur.val + " " : "null ");
            if (cur == null) {
                lastHead = curHead;
                while (lastHead != null && lastHead.left == null && lastHead.right == null) {
                    lastHead = lastHead.next;
                }
                
                // System.out.println("       " + lastHead.left + " " + lastHead.right + " " + lastHead.next.val);
                
                if (lastHead == null) {
                    break;
                }
                curHead = lastHead.left != null ? lastHead.left : lastHead.right;
                
                last = lastHead;
                cur = curHead;
                if (cur != null) {
                    System.out.println(lastHead.val + " " + curHead.val);    
                }                
            }
        }        
        return root;        
    }
}