【两次过】Lintcode 453. 将二叉树拆成链表

将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说，用二叉树的 right指针，来表示链表中的 next 指针。

样例

              1
               \
     1          2
    / \          \
   2   5    =>    3
  / \   \          \
 3   4   6          4
                     \
                      5
                       \
                        6

挑战

不使用额外的空间耗费。

注意事项

不要忘记将左儿子标记为 null，否则你可能会得到空间溢出或是时间溢出。

---

解题思路：

万能的分治法。假装左子树与右子树都变成了链表形式，只需要将左子树连接上右子树即可，这个连接过程需要知道子树的最后一个节点，所以我们在递归时返回值就设为当前节点链表的最后一个节点。

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /**
     * @param root: a TreeNode, the root of the binary tree
     * @return: nothing
     */
    public void flatten(TreeNode root) {
        // write your code here
        helper(root);
    }
    
    //返回的是当前节点前序遍历的最后一个节点，方便最后进行左右子树连接
    private TreeNode helper(TreeNode root){
        if(root == null)
            return null;
        
        TreeNode lastLeft = helper(root.left);
        TreeNode lastRight = helper(root.right);
        
        //将左子树链表连接上右子树链表
        if(lastLeft != null){
            lastLeft.right = root.right;
            root.right = root.left;
            root.left = null;
        }
        
        if(lastRight != null)
            return lastRight;
            
        if(lastLeft != null)
            return lastLeft;
        
        return root;
    }
}