剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
//先序遍历第一个确定根节点，根据这个根节点在中序中查找，可以将中序列表分成左右两个子树，
//分别统计左子树和右子树节点数量，根据数量得出先序的左子树右子树分界点，进而得到左子树右子树的
//先序列表，回到了起始的条件，因此可以开始递归左子树右子树进行建立
//使用map存储中序列表，val是对应的下标值，快速查询下标值
//递归返回的情况有：
//1.pre==end返回该子树的根节点
//2.pre>end 返回null
//3.pre<end 从map中获取左子树右标和右子树左标，递归调用，为当前根节点添加构成的左子树和右子树
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder == null || preorder.length==0){
            return null;
        }
        Map<Integer,Integer> map = new HashMap();
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i],i);
        }
        TreeNode root = new TreeNode(preorder[0]);
        return buildTree(preorder,0,preorder.length-1,inorder,0,inorder.length-1,map);

    }

    public TreeNode buildTree(int[] preorder,int preLeft, int preRight,int[]inorder,int inLeft,int inRight ,Map<Integer,Integer> map){
      
        if (preLeft > preRight){
            return null;
        }
          //每一个子树的根节点都是先序的第一个
        TreeNode root = new TreeNode(preorder[preLeft]);
        if(preLeft == preRight){
            return root;
        }else{
            int midIndex = map.get(root.val);//中序列表根节点下标
            int leftNodes = midIndex-inLeft;//左子树节点数
            int rightNodes = inRight-midIndex; //右子树结点数
            TreeNode leftTree = buildTree(preorder,preLeft+1,preLeft+leftNodes,inorder,inLeft,midIndex-1,map);
            TreeNode rightTree = buildTree(preorder,preRight-rightNodes+1,preRight,inorder,midIndex+1,inRight,map);
            root.left = leftTree;
            root.right = rightTree;
            return root;
        }

    }
}