2021-03-02

本题可以先不急着使用代码实现，可以试着在纸上看看怎么还原，不难发现，前序遍历的第一个数3，可以把中序遍历划分为两部分，左边是左子树，右边是右子树。3则是根节点。而左右子树又可以用同样的方法继续划分，所以可以使用分治思想来解题。为了减少遍历，可以将中序遍历放在map里，这样可以在时间复杂度为O(1)中计算得出前序遍历某一个数在中序遍历数组中的下标。
下图是计算前序遍历左子树范围和中序遍历左子树范围，以及右子树范围… … 方便将它们当做完整的子树的前中序遍历传入递归子函数。

class Solution {
    public static TreeNode buildTree(int[] preorder, int[] inorder) {
    	// 预处理
        if (preorder.length == 0){
            return null;
        }
        if (preorder.length == 1){
            return new TreeNode(preorder[0]);
        }
        
        int n = preorder.length - 1;
        // 构建map
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < inorder.length; i++){
            map.put(inorder[i], i);
        }
		// 开始递归
        TreeNode treeNode = f(0, n, 0, n, preorder, inorder, map);
        return treeNode;
    }
    public static TreeNode f (int preleft, int preright, int inleft, int inright, int[] preorder, int[] inorder, Map<Integer, Integer> map){
        if (preleft > preright || inleft > inright){
            return null;
        }
        // 根节点一定是preorder范围内的最左边
        TreeNode head = new TreeNode(preorder[preleft]);
        // 计算左子树的范围
        int index = map.get(preorder[preleft]);
        int prel = preleft + 1;
        int prer = preleft + index - inleft ;
        int inl = inleft;
        int inr = index - 1;
        head.left = f(prel, prer, inl, inr, preorder, inorder, map);
        // 计算右子树的范围
        prel = prer + 1;
        prer = preright;
        inl = index + 1;
        inr = inright;
        head.right = f(prel, prer, inl, inr, preorder, inorder, map);
        return head;
    }
}