积沙成塔
题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
算法分析:
在根据前序遍历和中序遍历生成二叉树的情景中,首先通过前序遍历得到根,再根据中序遍历得到左右子树的元素,再按照同样的思路继续生成左右子树。从上边可以看出这是一个递归的过程,因此实现思路是在确定左右子树的前提下递归创建子树。
实现代码:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class CreatTree {
public TreeNode reConstructBinaryTree(int [] pre, int [] in) {
if(pre.length==1)
return new TreeNode(pre[0]);
if(pre.length==0)
return null;
int index=0;
TreeNode gen=new TreeNode(pre[0]);
while(in[index]!=pre[0])
index++;//确定根
int[] inLeft=new int[index];//确定左子树中序
int[] preLeft=new int[index];//确定左子树前序
for(int i=0;i<index;i++){
inLeft[i]=in[i];
preLeft[i]=pre[i+1];
}
int[] inRight=new int[in.length-index-1];//确定右子树中序
int[] preRight=new int[in.length-index-1];//确定右子树前序
for(int i=index+1;i<in.length;i++){
inRight[i-index-1]=in[i];
preRight[i-index-1]=pre[i];
}
gen.left=reConstructBinaryTree(preLeft,inLeft);
gen.right=reConstructBinaryTree(preRight,inRight);
return gen;
}
}