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

7-23 还原二叉树 (25分)
给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

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

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

输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5

#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <algorithm>
using namespace std;
char in[1005], pre[1005];
int dfs(char *pre, char *in, int n)
{
    if (n == 0)
        return 0;
    int i;
    for (i = 0; i < n; i++)
    {
        if (in[i] == *pre)
        {
            break;
        }
    }
    int left = dfs(pre + 1, in, i);
    int right = dfs(pre + i + 1, in + i + 1, n - i - 1);
    return max(left, right) + 1;
}
int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> pre[i];
    }
    for (int i = 0; i < n; i++)
    {
        cin >> in[i];
    }
    cout << dfs(pre, in, n);
    return 0;
}