给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。

输入格式:

输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。

输出格式:

输出为一个整数,即该二叉树的高度。

输入样例:

9
ABDFGHIEC
FDHGIBEAC
输出样例:

5

#include<stdio.h>
#include<stdlib.h>
#include<math.h>
typedef char ElemType;
typedef struct BiTreeType{
	ElemType data;
	struct BiTreeType *lchild,*rchild;
}BiTreeNode,*BiTree;
BiTree PreInorder(char preord[],char inord[],int i,int j,int k,int h)
{
	BiTree t;
	int m;
	t=(BiTreeNode *)malloc(sizeof(BiTreeNode));
	t->data=preord[i];
	m=k;
	while(inord[m]!=preord[i])
	 m++;
	if(m==k)
	 t->lchild=NULL;
	else
	 t->lchild=PreInorder(preord,inord,i+1,i+m-k,k,m-1);
	if(m==h)
	 t->rchild=NULL;
	else
	 t->rchild=PreInorder(preord,inord,i+m-k+1,j,m+1,h);
	return t;
}
BiTree CreateBiTree(char preord[],char inord[],int n)
{
	BiTree root;
	if(n<=0)
	 root=NULL;
	else
	 root=PreInorder(preord,inord,0,n-1,0,n-1);
	 return root;
}
int BitreeDepth(BiTree root)
{
	int d=0,depthl,depthr;
	if(root==NULL)
	 return 0;
	depthl=BitreeDepth(root->lchild);
	depthr=BitreeDepth(root->rchild);
	if(depthl>depthr)
	 d=depthl;
	else
	 d=depthr;
	return d+1;
}
int main()
{
	int n;
	BiTree root;
	scanf(\"%d\",&n);
	char preord[50],inord[50];
	scanf(\"%s\",preord);
	getchar();
	scanf(\"%s\",inord);
	root=CreateBiTree(preord,inord,n);
	n=BitreeDepth(root);
	printf(\"%d\",n);
	return 0;
}

收藏 打印