[LeetCode系列]Populating Next Right Pointers in Each Node II

将会陆续放出本人刷LeetCode所见到的一些比较有趣或者不会的题目的随笔。

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 I

两题有一些有趣的区别，第二题是非完全二叉树。所以在做第一题时用了队列对每一层的每个节点进行连接，但在这一题里面就无法简单地通过这个方法去实现。

一开始我同样用队列分层去实现，第i层的满节点数是Ki=2^(depth-1)，实际节点数为Si，但因为是非完全二叉树，所以通过计算K_i-1-S_i-1，那么第i层的节点数应该是K_i-(K_i-1-S_i-1)-T, T为i-1层非空节点中left chile 或者 right child为空的个数。但用这个方法最后大数无法通过，暂时还没找出BUG。望哪位大神可以提供一个详细的队列方法的解题思路。

接下去是递归方法。

考虑的Point在每个节点的next节点可以在横向上找到存在的那个节点，无论是父节点next节点的左或者右孩子，又或者是父节点next的next节点的左或者右节点。

注意需要优先找出right child的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;
      TreeLinkNode p = root.next;
      while (p != null) {
          if (p.left != null) {
              p = p.left;
              break;
          } else if (p.right != null) {
              p = p.right;
              break;
          }
          p = p.next;
      }
      
      if (root.right != null) {
          root.right.next = p;
      } 
      
      if (root.left != null) {
          root.left.next = root.right == null?p:root.right;
      }
      
      connect(root.right);
      connect(root.left);
    }

}