重建二叉树
题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出二叉树并输出他的根节点。二叉树的定义如下:
public static class BinaryTreeNode{
int value;
BinaryTreeNode left;
BinaryTreeNode right;
}
思路
在二叉树的前序遍历中,第一个数字总是树的根节点。但在中序遍历中,根节点的值在序列的中间。左子树的值位于根节点的左侧,而右子树的节点位于根节点值的右边。所以我们要扫描中序遍历序列,才能找到根节点的值。
如示例中所示,1为整颗树的根节点,所以1左边的三个数字都是左子树的值,右边都是右子树的值。同理在247,472中,2为根节点,47都是以2为根节点的左子树的值,如此递归可求。
代码
public BinaryTreeNode reConstruct(int[] preOrder, int[] inOrder){
if(preOrder==null||inOrder==null||preOrder.length!=inOrder.length||preOrder.length<=0)
return null;
return construct(preOrder,0,preOrder.length-1,inOrder,0,inOrder.length-1);
}
public static BinaryTreeNode construct(int[] preOrder,int pstart,int pend,int[] inOrder,int,istart,int iend){
if(pstart>pend||istart>iend)
{
return null;
}
int value=preorder[pstart];
int index=istart;
while(index<iend && inOrder[index]!=value){
index++;
}
BinaryTreeNode root=new BinaryTreeNode();
root.value=value;
root.left=construct(preOrder, pstart+1,pstar+index-istart, inOrder, is,index-1);
root.left=construct(preOrder, pstar+index-istart+1,pend, inOrder, index+1,iend);
return node;
}
//中序遍历递归打印
public static void printTree(BinaryTreeNode node){
if(node!=null){
printTree(node.left);
System.out.print(node.value+" ");
printTree(node.right);
}
}