【Lintcode】453. Flatten Binary Tree to Linked List

题目地址：

https://www.lintcode.com/problem/flatten-binary-tree-to-linked-list/description

给定一棵二叉树，返回其先序遍历形成的链表。要求直接修改二叉树，以右孩子作为链表的next节点。

容易想到的是，先递归求解左右子树，再将左子树的尾巴节点连到右子树之前，最后再将根连到左子树之前，最后将左子树清为null即可。由于我们需要频繁求解左子树flatten后的最后一个节点，所以可以干脆设置一个函数，功能是，flatten以root为根的子树，并返回其最右边的节点。代码如下：

public class Solution {
    /**
     * @param root: a TreeNode, the root of the binary tree
     * @return: nothing
     */
    public void flatten(TreeNode root) {
        // write your code here
        flattenAndGetlast(root);
    }
    // 功能是flatten以root为根的子树，并返回flatten后的子树的最右边的节点
    private TreeNode flattenAndGetlast(TreeNode root) {
    	// 如果是空树，则返回null
        if (root == null) {
            return null;
        }
        // 接下来分而治之
        TreeNode leftLast = flattenAndGetlast(root.left);
        TreeNode rightLast = flattenAndGetlast(root.right);
        // 接下来将左子树的链表插在根与右子树之间，方法和在链表中插入是一样的
        // 需要注意的是，如果左子树为空，这一步就不用做了；
        if (leftLast != null) {
            leftLast.right = root.right;
            root.right = root.left;
            root.left = null;
        }
        // 接下来需要返回树中最右边的节点
        // 先考虑右子树为空的情况，如果左子树也为空，则返回根，否则返回左子树的最右节点；
        // 如果右子树不为空，那当然就返回右子树最右节点
        if (rightLast == null) {
            return leftLast == null ? root : leftLast;
        } else {
            return rightLast;
        }
    }
}

时间复杂度$O(n)$，空间$O(h)$。