leetcode：从前序与中序遍历序列构造二叉树（java，考察点：构造二叉树）

题目

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

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

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

思路

1. 前序遍历：每一个节点都是根节点
2. 中序遍历：根节点左边的为左子树，根节点右边的为右子树
3. 遍历前序数组找到根节点，然后遍历中序数组找到在左边还是右边

java

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder == null) {
            return null;
        }
        
        return buildCore (preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
    }
    
    private TreeNode buildCore(int[] preorder, int preStr, int preEnd, int[] inorder, int inStr, int inEnd) {
        int rootValue = preorder[preStr];
        TreeNode root = new TreeNode(rootValue);
        
        if (preStr == preEnd) {
            return null;
        }
        
        int rootInorder = inStr;
        while (inorder[rootInorder] != rootValue &&  rootInorder <= inEnd) {
            rootInorder++;
        }
        
        int leftLength = rootInorder - inStr;
        int leftPreEnd = preStr + leftLength;
        
        if (leftLength > 0) {
            root.left = buildCore(preorder, preStr + 1, leftPreEnd, inorder, inStr, inEnd);
        }
        
        if (leftLength < preEnd - preStr) {
            root.right = builderCore(preorder, leftPreEnd + 1, preEnd, inorder, rootInorder + 1, inEnd);
        }
        
        return root;
    }
}