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

和这个系列第一道题比较起来，少了一个完全二叉树的条件，那么就需要用迭代的方法去找next。

有一个注意事项是，要先做右子树，再作左子树，考虑以下情况：

1. l1和r1分别为root节点的两个子节点，如果说假设我们先做l1

2. 做到l1的右子节点的时候，需要到r1的子节点里面去找next，这时候如果r1的两个子节点都是空，那么需要继续到r1的next中去找，这时候因为我们先递归了l1，r1的next还没有被赋值，所以会出现丢失next的情况。

/**
 * Definition for binary tree with next pointer.
 * public class TreeLinkNode {
 *     int val;
 *     TreeLinkNode left, right, next;
 *     TreeLinkNode(int x) { val = x; }
 * }
 */
public class Solution {
    public void connect(TreeLinkNode root) {
        if(root == null){
            return;
        }
        if(root.right!=null){
            root.right.next = findNext(root.next);
        }
        if(root.left!=null){
            root.left.next = root.right==null?findNext(root.next):root.right;
        }
        connect(root.right);
        connect(root.left);
    }
    public TreeLinkNode findNext(TreeLinkNode root){
        if(root==null){
            return null;
        }else{
            TreeLinkNode iter = root;
            TreeLinkNode result = null;
            while(iter!=null){
                if(iter.left!=null){
                    result = iter.left;
                    break;
                }
                if(iter.right!=null){
                    result = iter.right;
                    break;
                }
                iter = iter.next;
            }
            return result;
        }
    }
}