数据结构实验之二叉树四:(先序中序)还原二叉树
Time Limit: 1000MS
Memory Limit: 65536KB
Problem Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
Output
输出一个整数,即该二叉树的高度。
Example Input
9 ABDFGHIEC FDHGIBEAC
Example Output
5
代码如下:
#include<bits/stdc++.h>
using namespace std;
struct node
{
char data;
struct node *left;
struct node *right;
};
struct node *Creat(int n,char *str,char *str1)
{
struct node *tree;
if(n==0)
return NULL;
tree=new node;
char *p;
tree->data=str[0];
for(p=str1; p!='\0'; p++)
{
if(*p==str[0])
break;
}
int t=p-str1;
tree->left=Creat(t,str+1,str1);
tree->right=Creat(n-t-1,str+1+t,p+1);
return tree;
};
int Deepth(struct node *tree)
{
int H_left;
int H_right;
int max;
if(tree)
{
H_left=Deepth(tree->left);
H_right=Deepth(tree->right);
if(H_left>=H_right)
max=H_left;
else
max=H_right;
return max+1;
}
else
return 0;
}
int main()
{
int n;
char str[100];
char str1[100];
struct node *tree;
while(cin>>n)
{
cin>>str;
cin>>str1;
tree=Creat(n,str,str1);
int m=Deepth(tree);
cout<<m<<endl;
}
return 0;
}