重建二叉树(已知中序前序)
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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 右子树的中序遍历是5386
然后左子树的前序遍历是247,这怎么知道呢,其实就是根据左子树中序遍历的长度可以知道左子树前序遍历的长度。
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length==0)return null;
return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
}
public TreeNode build(int[] preorder,int lo1,int hi1, int[] inorder,int lo2,int hi2){
if(lo1>hi1 || lo2>hi2)return null;
int index=lo2;
for(int i=lo2;i<=hi2;i++){
if(inorder[i]==preorder[lo1]){
index=i;
break;
}
}
int leftsize=index-lo2;
TreeNode root=new TreeNode(preorder[lo1]);
root.left=build(preorder,lo1+1,lo1+leftsize,inorder,lo2,index-1);
root.right=build(preorder,lo1+leftsize+1,hi1,inorder,index+1,hi2);
return root;
}
}