剑指 07 java 重建二叉树

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

例如，给出

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

3

/
9 20
/
15 7

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //用map来存中序遍历的数组信息，键为数组的值，值为数组的索引号，因为需要通过前序得到中序根节点的位置
    Map<Integer,Integer> map = new HashMap();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder == null||inorder == null|| preorder.length<=0||inorder.length<=0){
            return null;
        }
        for(int i = 0;i<inorder.length;i++){
            map.put(inorder[i],i);
        }
        return Change(preorder,0,0,inorder.length-1);
    }
    //由于要用递归所以肯定要写一个递归方法，方法返回值是TreeNode
    //参数：递归中先要知道根节点的位置，所以需要根位置索引(从前序遍历得到)，还需要树的左右边界（从中序遍历得到）
    //还需要前序遍历的数组，因为中序遍历已经存在了MAP中，所以不需要传中序遍历
    public TreeNode Change(int[] preorder,int head,int left,int right){
        //一旦左边界大于又边界就要返回null
        if(left > right){
            return null;
        }
        //先声明根节点
        TreeNode root = new TreeNode(preorder[head]);
        //获取根节点在中序遍历中的位置
        int a = map.get(preorder[head]);
        //左子树都在根节点位置的左边,左子树的根节点就是head+1,边界是left，右边界是a-1
        root.left = Change(preorder,head +1,left,a-1);
        //右子树都在根节点位置的右边,右子树的根节点就是根节点位置加上左子树长度+1,左边界是a+1，右边界是right
        root.right = Change(preorder,a+head-left+1,a+1,right);
        return root;

    }
}