leetcode—105. 从前序与中序遍历序列构造二叉树

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

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

例如，给出

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

     3    
    / \  
   9  20
     /  \    
    15   7

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return build(preorder,inorder,0,preorder.length-1,0,inorder.length-1);
    }

    public TreeNode build(int[] preorder, int[] inorder,int prestart,int preend,int instart,int inend) {
        if(prestart>preend || instart>inend){
            return null;
        }
        if(prestart==preend){
            return new TreeNode(preorder[prestart]);
        }

        int root=preorder[prestart];
        TreeNode rootNode=new TreeNode(root);
        int i=instart;
        for(;i<=inend;i++){
            if(inorder[i]==root){
                break;
            }
        }
        int leftLen=i-instart;
        int rightLen=inend-i;

        rootNode.left=build(preorder,inorder,prestart+1,prestart+leftLen,instart,i-1);
        rootNode.right=build(preorder,inorder,prestart+leftLen+1,preend,i+1,inend);

        return rootNode;
    }
}