leetcode【每日一题】105. 从前序与中序遍历序列构造二叉树 Java

我的leetcode的答案都已经上传到我的git仓库https://github.com/ragezor/leetcode

题干

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

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

例如，给出

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

    3
   / \
  9  20
    /  \
   15   7

想法

显然是使用递归嘛。
我们首先可以确定的是前序的第一个点肯定是根节点，那么在中序里找到根节点的位置。
此位置之前的都是左子树，之后的都是右子树。
那么就可以唯一确定的，也就是说可以递归了。
最愚蠢的办法，每次都到整个中序去找，但是显然不ok。
因为区间很容易确定
画个图：

可以得到左右子树分别在两个序列的区间，那么代码就好写了。

Java代码

package daily;

public class BuildTree {
    public static class TreeNode {
      int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
 }
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        //序列长度，两个一样所以只用求一个
        int prelen=preorder.length;
    return  rebuild(preorder,0,prelen-1,inorder,0,prelen-1);
    }
    /*
     * @Description:
     * @Author: ragezor
     * @Date: 2020/5/22 10:52 上午
     * @Param [preorder, prestart在前序的开始, preend在前序的结束, inorder, instart中序的开始, inend中序的结束] 
     * @return daily.BuildTree.TreeNode
     **/
    public TreeNode rebuild(int []preorder,int prestart,int preend,int[] inorder,int instart,int inend){
        //空判断
        if(prestart>preend||instart>inend){
            return  null;
        }
        //现在的节点
        TreeNode root=new TreeNode(preorder[prestart]);
        for (int i=inend;i>=instart;i--){
            if(inorder[i]==preorder[prestart]){
                //左子树长度
                int len=i-instart;
                //看图对应的两个序列的区间很好懂
                root.left=rebuild(preorder, prestart+1, prestart+len, inorder, instart,i-1);
                root.right=rebuild(preorder, prestart+len+1, preend, inorder, i+1,inend);
                break;


            }
        }
        return  root;
    }


}

我的leetcode的答案都已经上传到我的git仓库https://github.com/ragezor/leetcode