leetcode:105. 从前序与中序遍历序列构造二叉树(java)

package LeetCode;

import java.util.ArrayList;
import java.util.List;

/*
105. 从前序与中序遍历序列构造二叉树
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如，给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：
    3
   / \
  9  20
    /  \
   15   7
 */
public class BuildTree {

    public TreeNode buildTree(int[] preorder, int[] inorder) {

        return dfs(0, preorder.length - 1, 0, inorder.length - 1, preorder, inorder);
    }

    public TreeNode dfs(int ps, int pe, int is, int ie, int[] preorder, int[] inorder) {
        if (ps > pe) return null;
        TreeNode r = new TreeNode(preorder[ps]);
        int mid = -1;
        for (int i = 0; i <= ie; i++) {
            if (inorder[i] == preorder[ps]) {
                mid = i;
                break;
            }
        }
        if (mid == -1) return null;
        //左子树的个数
        //前序遍历[ps][ps+1...ps+c][ps+c+1...pe]
        //后序遍历[is...mid-1][mid][mid+1...ie]
        int c = mid - is;
        r.left = dfs(ps + 1, ps + c, is, mid - 1, preorder, inorder);
        r.right = dfs(ps + c + 1, pe, mid + 1, ie, preorder, inorder);
        return r;
    }
}