数据结构实验之二叉树四:(先序中序)还原二叉树
Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
Output
输出一个整数,即该二叉树的高度。
Sample
Input
9 ABDFGHIEC FDHGIBEAC
Output
5
代码:
#include<bits/stdc++.h>
using namespace std;
typedef struct Tnode
{
char data;
struct Tnode *lchild, *rchild;
}Tnode, *Tlink;
Tlink createtree(char *pre,char *in, int l)
{
if(l <= 0)
return NULL;
Tlink r = new Tnode;
r->data = *pre;
char *t = in;
int i;
for(i = 0; i < l; i++)
{
if(*(t+i) == *pre)
break;
}
t = t+i;
r -> lchild = createtree(pre+1,in,i);
r -> rchild = createtree(pre+i+1,in+i+1,l-i-1);
return r;
}
int deepth(Tlink t)
{
int m,n;
if(t == NULL) return 0;
else
{
m = deepth(t->lchild);
n = deepth(t->rchild);
if(m > n)
return m+1;
else
return n+1;
}
}
int main()
{
int n;
char s1[55],s2[55];
while(~scanf("%d",&n))
{
scanf("%s",s1);
scanf("%s",s2);
Tlink t =createtree(s1,s2,n);
cout << deepth(t) << endl;
}
return 0;
}