剑指Offer（4）：重建二叉树

题目

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

思路

前序遍历：中左右
中序遍历：左中右
后序遍历：左右中
本题为前序遍历和中序遍历，最少需要两种遍历方式，才能重建二叉树。
前序遍历序列中，第一个数字总是树的根结点的值。根据该值找到根节点在中序遍历中的位置，之后划分出左右子树的前序、中序遍历数组，递归处理。

代码

import java.util.ArrayList;

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre==null||in==null||pre.length==0||in.length==0||pre.length!=in.length)
            return null;
        ArrayList<Integer>PRE, IN;
        PRE=new ArrayList<>();
        IN=new ArrayList<>();
        for(int i=0;i<pre.length;i++){
            PRE.add(pre[i]);
            IN.add(in[i]);
        }
        return helper(PRE, IN);
    }
    
    private TreeNode helper(ArrayList<Integer> pre, ArrayList<Integer> in){
        if(pre.size()==0)
            return null;
        int rootVal = pre.get(0);
        TreeNode root = new TreeNode(rootVal);
        int mid=0;
        while(mid<in.size()&&in.get(mid)!=rootVal){
            mid++;
        }
        ArrayList<Integer> preL,preR,inL,inR;
        preL = new ArrayList<>();
        preR = new ArrayList<>();
        inL = new ArrayList<>();
        inR = new ArrayList<>();
        for(int i=0;i<mid;i++){
            preL.add(pre.get(i+1));
            inL.add(in.get(i));
        }
        for(int i=mid+1;i<pre.size();i++){
            preR.add(pre.get(i));
            inR.add(in.get(i));
        }
        root.left = helper(preL, inL);
        root.right = helper(preR, inR);
        return root;
    }
}