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

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

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,mid+k+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;
}