7-23 还原二叉树 (25分) 【PTA 先序 + 中序 求 高度】

7-23 还原二叉树 (25分)

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

输入格式:

输入首先给出正整数N（≤50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include <iostream>
#include<bits/stdc++.h>

using namespace std;
struct node
{
    char data;
    struct node *lc,*rc;
};
char s[1000];
char t[1000];
int n;
struct node *create(int len, char a[], char b[])
{
    if(len == 0) return NULL;
    int i;
    struct node *root;
    root = new node;
    root -> data = a[0];
    for(i = 0; i < len; i ++)
    {
        if(a[0] == b[i])
        {
            break;
        }
    }
    root -> lc = create(i,a+1,b);
    root -> rc = create(len - i - 1, a + i + 1, b + i + 1);
    return root;
};

int fin(struct node * root)
{
    int d1,d2;
    int h = 0;
    if(root)
    {
        d1 = fin(root -> lc);
        d2 = fin(root -> rc);
        h = max(d1+1,d2+1);
    }
    return h;
}
int main()
{
//    int n;
    scanf("%d", &n);
    scanf("%s %s",s,t);
    struct node *root;;
    root = create(n,s,t);
    int ans = fin(root);
    printf("%d\n",ans);
    return 0;
}