题解 | #重建二叉树#

import java.util.*;
import java.util.HashMap;
import java.util.Map;
/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {

        if(pre==null||vin==null||pre.length != vin.length){
            return null;
        }


        TreeNode head = TreeFun(pre,0,pre.length-1,vin,0,vin.length-1);
        return head;

        
    }

    public static TreeNode TreeFun(int[] pre, int l1,int r1,int[] vin,int l2,int r2){     
        if(l1>r1){
            return null;

        }

        TreeNode head = new TreeNode(pre[l1]);
        if(l1==r1){
            return head;
        }

        int find = l2;
        while(vin[find] != pre[l1]){
            find++;
        }

        head.left = TreeFun(pre,l1+1,find-l2+l1,vin,l2,find-1);
        head.right = TreeFun(pre,find-l2+l1+1,r1,vin,find+1,r2);
        return head;


    }
}