leetcode-105 从前序与中序遍历序列构造二叉树（Medium）

题目描述

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

给定代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        
    }
}

题解一

根据前序遍历，可以知道第一个元素是根节点。
根据中序遍历，可以获取根节点的下标，下标左边的子区间就是左子树，下标右边的子区间就是右子树。

这是一个递归的过程，每次确定前序遍历和中序遍历的范围，就能构造一个子树。
在中序遍历寻找根节点的过程可以使用Map做索引。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int [] pre;
    private int [] in;
    private HashMap<Integer, Integer> map = new HashMap<>();

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for (int i = 0 ; i < inorder.length ; i++) {
            map.put(inorder[i], i);
        }
        pre = preorder;
        in = inorder;
        return deep(0, preorder.length - 1, 0, preorder.length - 1);
    }



    private TreeNode deep(int start, int end, int start2, int end2)
    {
        if (start > end || start2 > end2) {
            return null;
        } else {
            TreeNode result = new TreeNode(pre[start]);
            int index = map.get(pre[start]);
            int leftLength = index - start2;
            int rightLength = end2 - index;
            if (leftLength > 0)
                result.left = deep(start + 1, start + 1 + leftLength, start2, index - 1);
            if (rightLength > 0)
                result.right = deep(start + leftLength + 1, end, index + 1, end2);
                    
            return result;
        }
    }
}

时间复杂度：$O(n)$，其中 $n$ 是树中的节点个数。

空间复杂度：$O(n)$，除去返回的答案需要的 $O(n)$ 空间之外，我们还需要使用 $O(n)$ 的空间存储哈希映射，以及 $O(h)$（其中 h 是树的高度）的空间表示递归时栈空间。这里 $h<n$，所以总空间复杂度为$O(n)$。