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

题目：leetcode105. 从前序与中序遍历序列构造二叉树

描述

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

思路

这道题跟106是一样的，只是把后序遍历改成了前序遍历，中序遍历时肯定要的，没有中序遍历时得不到二叉树的，所以我们还是根据前序遍历序列的第一个数字来分割中序遍历序列。

代码

import java.util.HashMap;
import java.util.Map;

public class Solution {
    Map<Integer,Integer> map;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        map=new HashMap<>();
        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i],i);
        }
        return createTree(preorder,0,preorder.length,inorder,0,inorder.length);
    }

    private TreeNode createTree(int[] preorder,int prebegin,int preend,int[] inorder,int inbegin,int inend)
    {
        if(prebegin>=preend || inbegin>=inend)
            return null;
        int rootIndex=map.get(preorder[prebegin]);
        TreeNode root=new TreeNode(inorder[rootIndex]);
        int lenleft=rootIndex-inbegin;
        root.left=createTree(preorder,prebegin+1,prebegin+1+lenleft,inorder,inbegin,inbegin+lenleft);
        root.right=createTree(preorder,prebegin+lenleft+1,preend,inorder,inbegin+lenleft+1,inend);
        return root;
    }
}