Leetcode105.从前序与中序遍历序列构造二叉树 Construct Binary Tree from Preorder and Inorder Traversal(Java)

Leetcode105.从前序与中序遍历序列构造二叉树 Construct Binary Tree from Preorder and Inorder Traversal(Java)

##Array##, ##Tree##, ##Depth-first Search##

从前序与中序遍历序列构造二叉树

本题采用递归做法，创建根节点，递归创建左右子树，根节点链接两棵子树

• 利用前序遍历的性质，前序遍历的第一个数就是根节点的值
• 在中序遍历中按根节点的值找到根节点的位置temp，则temp左边时左子树的中序遍历，右边是右子树的中序遍历
• 不妨设左子树中序遍历的长度为len1，则在前序遍历中，根节点后面的len1个数，是左子树的前序遍历，剩余右边的数是右子树的前序遍历
• 有了左右子树的前序遍历和中序遍历，便可以递归创建左右子树，最后与根节点相连
• 可以用哈希表将中序遍历的值和位置作映射，这样在中序遍历中招根节点时时间复杂度变为O(1)
• 一定考虑递归出口pl > pr || il > ir

时间复杂度： O(n)

class Solution {
    HashMap<Integer, Integer> hash = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for (int i = 0; i < preorder.length; i ++) hash.put(inorder[i], i);
        return build(preorder, inorder, 0, preorder.length - 1, 0, inorder.length - 1);
    }
    
    public TreeNode build(int[] p, int[] i, int pl, int pr, int il, int ir) {
        if (pl == pr) return new TreeNode(p[pr]); // 这行可有可无，加快程序运行时间
        if (pl > pr || il > ir) return null;
        int temp = hash.get(p[pl]);
        TreeNode res = new TreeNode(p[pl]);
        res.left = build(p, i, pl + 1, pl + temp - il , il, temp - 1);
        res.right = build(p, i, pl + temp - il + 1, pr, temp + 1, ir);
        return res;
    }
}