数据结构实验之二叉树四:(先序中序)还原二叉树
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
Output
输出一个整数,即该二叉树的高度。
Sample Input
9
ABDFGHIEC
FDHGIBEAC
Sample Output
5
Hint
Source
xam
#include <bits/stdc++.h>
using namespace std;
struct bitree
{
char data;
struct bitree *lc;
struct bitree *rc;
};
bitree *create(int n,char a[],char b[])
{
bitree *tree;
char *p;
if(!n)
tree = NULL;
else
{
tree = new bitree;
tree->data = a[0];
for(p=b; p!='\0'; p++)
{
if(*p==a[0])
break;
}
int t;
t = p-b;
tree->lc = create(t,a+1,b);
tree->rc = create(n-1-t,a+t+1,p+1);
}
return tree;
}
int depth(bitree *tree)
{
int d = 0;
if(tree)
{
int ld = depth(tree->lc);
int rd = depth(tree->rc);
d = ld>rd?ld+1:rd+1;
}
return d;
}
int main()
{
int n;
char a[51],b[51];
while(cin>>n)
{
bitree *tree;
cin>>a>>b;
tree = create(n,a,b);
cout<<depth(tree)<<endl;
}
return 0;
}