【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(n) O(n),空间 O ( h ) O(h) O(h)

你可能感兴趣的:(#,树,分治与堆)