从前序与中序遍历序列构造二叉树

---

import java.util.HashMap;

public class PreInToTree {
    public static TreeNode preInToTree(int[] pre,int[] in){
        if(pre==null||in==null){
            return null;
        }

        //map存储中序遍历序列，key节点值，value：节点坐标，用于查询头节点在中序序列中的坐标
        HashMap<Integer, Integer> map = new HashMap<>(8);
        for(int i=0;i<in.length;i++){
            map.put(in[i],i);
        }

        return preIn(pre,0,pre.length-1,in,0,in.length-1,map);
    }


    public static TreeNode preIn(int[] pre, int pl, int pr, int[] in, int il, int ir, HashMap<Integer,Integer> map){
        if(pl>pr){
            return null;
        }

        //前序序列第一个元素为头节点
        TreeNode head = new TreeNode(pre[pl]);

        //获取头在中序的坐标，进而得出划分子树的边界值
        int index = map.get(pre[pl]);

        //pl，pr，分别为前序序列的左右边界。il，ir，分别为中序遍历的左右边界，递归构造左右子树
        head.left = preIn(pre, pl+1, pl + index - il, in, il, index-1, map);
        head.right=preIn(pre,pl + index - il+1,pr,in,index+1,ir,map);
        return head;

    }
}