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

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

根据前序序列可以找到树根，然后根据树根在中序序列的位置，可以确定左右子树，然后将左右子树分别作为一棵树进入递归，每次记录下左右子树的深度，最后返回最大深度。

#include<iostream>
#include<string>
using namespace std;
int CalHeight(char a[],char b[],int n)
{
    int i;
    if(n==0)return 0;
    for(i = 0;i < n;i++)
    {
        if(b[i] == a[0])
        {
            break;
        }
    }
    int x = CalHeight(a+1,b,i)+1; //计算左子树的深度
    int y = CalHeight(a+i+1,b+i+1,n-i-1)+1;//计算右子树的深度

    return x>y?x:y;
}
int main()
{
    char a[101],b[101];
    int n;
    cin >> n;
    cin >> a >> b;
    int cnt = CalHeight(a,b,n);
    cout << cnt << endl;
    return 0;
}