算法题目-二叉树的重建

题目：

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
}
}