算法题目-二叉树的重建

题目：

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

分析：

对于二叉树，前序遍历是，根左右。{1,2,4,7,3,5,6,8}

中序遍历是，左根右。{4,7,2,1,5,3,8,6}

因此可知前序遍历中的第一个数1为根节点，再看中序遍历，可以得到1之前的为左子树{4,7,2}，1之后的为右子树{5,3,8,6}。

再继续进行递归分析即可。

解答：

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length==0||in.length==0||pre.length!=in.length){
            return null;
        }
        int i=0;
        TreeNode root=new TreeNode(pre[0]);
        while(in[i]!=root.val){
            i++;
        }
        int[] preLeft=new int[i];
        int[] inLeft=new int[i];
        int[] preRight=new int[pre.length-1-i];
        int[] inRight=new int[in.length-1-i];
        for(int j=0;j<pre.length;j++){
            if(j<i){
                preLeft[j]=pre[j+1];
                inLeft[j]=in[j];
            }else if(j>i){
                preRight[j-i-1]=pre[j];
                inRight[j-i-1]=in[j];
            }
        }
        root.left=reConstructBinaryTree(preLeft,inLeft);
        root.right=reConstructBinaryTree(preRight,inRight);
        return root;
    }
}