输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
记前序遍历pre[],中序遍历in[].
先获取当前前序遍历数组pre的第一个元素记作a,比如遍历到i位置的值为K,在中序遍历in的(start-end区间)中找到和k值相等的元素,所在的位置为L。
则以i为父节点的左边子树的元素在in中 (start,i-1),左子树中前序遍历的第一个元素位置则为a+1;
以i为父节点的右边子树的元素在in中(i+1,end),右子树中前序遍历的第一个元素位置则为a+1+(i-start)
我们只需要通过递归重复两步就能将树构造出来。
需要注意的是递归的结束条件是:
左边节点left>=right或者中序遍历的第一个元素超出了数组范围。
public TreeNode reConstructBinaryTree(int[] pre,int[] in) {
return construct(pre,in,0,0,in.length-1);
}
public TreeNode construct(int[] pre,int[] in,int prestart,int instart,int inend){
if(prestart>pre.length||instart>inend){
return null;
}else if(instart==inend){
return new TreeNode(in[instart]);
}
int middle = pre[prestart];
int temp = -1;
for(int i=instart;i<=inend;i++){
if(in[i]==middle){
temp=i;
break;
}
}
TreeNode t = new TreeNode(middle);
t.left=construct(pre,in,prestart+1,instart,temp-1);
t.right=construct(pre,in,prestart+1+(temp-instart),temp+1,inend);
return t;
}