数据结构实验之二叉树四：还原二叉树

数据结构实验之二叉树四：还原二叉树

Time Limit: 1000MS Memory limit: 65536K

题目描述

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

输入

输入数据有多组，每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数，随后2行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母(区分大小写)的字符串。

输出

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

示例输入

9
ABDFGHIEC
FDHGIBEAC

示例输出

5

提示

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std;
struct node
{
char data;
struct node *l, *r;
};
struct node *creat(char *pre,char *mid,int len)
{
if(len <= 0){
return NULL;
}
struct node *root;
root = (struct node *)malloc(sizeof(struct node));
root->data = *pre;
char *p;
for(p = mid;;p++){
if(*p == *pre){
break;
}
}
int k = p-mid;
root->l = creat(pre+1,mid,k);
root->r = creat(pre+k+1,p+1,len-k-1);
return root;
}
int deep(struct node *tree)
{
int h, k;
if(tree){
h = deep(tree->l);
k = deep(tree->r);
return max(h,k)+1;
}else {
return 0;
}
}
int main()
{
char s1[100];
char s2[100];
int n;
struct node *tree;
while(~scanf("%d", &n)){
scanf("%s %s", s1, s2);
int len = strlen(s1);
tree = creat(s1,s2,len);
printf("%d\n", deep(tree));
}

return 0;
}