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

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

Time Limit: 1000MS Memory Limit: 65536KB

Submit Statistic

Problem Description

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

Input

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

Output

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

Example Input

9

ABDFGHIEC

FDHGIBEAC

Example Output

5

Hint

大言不惭的说说水题。。虽然做的过程并不一帆风顺

#include <stdio.h>
#include <stdlib.h>

typedef struct node
{
char data;
struct node *l,*r;
}node;

node *create(int len,char *pre,char *in)//老套路了，知道先序和中序搞出整棵树
{
int i;
node *root;
if(len<=0)
return NULL;
else{
root = (node *)malloc(sizeof(struct node));
char x = pre[0];
root->data = x;
for(i=0; i<len; i++){
if(in[i]==x)
break;
}
root->l = create(i,pre+1,in);
root->r = create(len-i-1,pre+i+1,in+i+1);
}
return root;
}

int deep(node *root)//求深度的话，就看左右子树哪个更深就返回哪个的深度，最后加个1就是整棵树的深度了
{
int d = 1;
if(root){
int dl = deep(root->l);
int dr = deep(root->r);
d = dl>dr?dl+1:dr+1;
}
else
return 0;//一定到最后没有左右子树的话要返回0，这样就不会又加1了
return d;
}

int main()
{
int len;
char pre[100],in[100];
while(~scanf("%d",&len)){
scanf("%s %s",pre,in);
node *root = create(len,pre,in);
printf("%d\n",deep(root));
}
return 0;
}