[LeetCode][Java] Populating Next Right Pointers in Each Node II

题目：

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

Note:

• You may only use constant extra space.

For 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

题意：

伴随着问题《Populating Next Right Pointers in Each Node》

如果给定的这颗树是任意类型的树，你之前的解法是否依然可行呢？

注意：

    你只能使用常数空间。

比如

给定如下的二叉树

         1
       /  \
      2    3
     / \    \
    4   5    7

调用你的函数之后，这颗二叉树应该是如下的样子：

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

算法分析：

 * 整体思想与《Populating Next Right Pointers in Each Node》相同，同样是递归

 * 但是在相邻节点为空的情况下，需要去寻找不为空的第一个节点，并与之相连

AC代码：

<span style="font-family:Microsoft YaHei;font-size:12px;">public class Solution 
{
    public void connect(TreeLinkNode root) 
    {  
        if (root == null) return;  
          
        //如果右孩子不为空，左孩子的next是右孩子。  
        //反之，找root next的至少有一个孩子不为空的节点  
        if (root.left != null) 
        {  
            if (root.right != null)
            {  
                root.left.next = root.right;  
            }  
            else
            {  
                TreeLinkNode p = root.next;  
                while (p != null && p.left == null && p.right == null)  
                    p = p.next;  
                if (p != null)  
                    root.left.next = p.left == null ? p.right : p.left;  
            }  
        }  
          
        //右孩子的next 根节点至少有一个孩子不为空的next  
        if (root.right != null)
        {  
            TreeLinkNode p = root.next;  
            while (p != null && p.left == null && p.right == null)  
                p = p.next;  
            if (p != null)  
                root.right.next = p.left == null ? p.right : p.left;  
        }  
        connect(root.right);      
        connect(root.left);  
    }  
}</span>