还原二叉树

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

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

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include<bits/stdc++.h>
using namespace std;
int dfs(char* pre,char* in,int n)   //求二叉树的高度
{
    if(n == 0)    //若没有结点，为空树
    {
        return 0;
    }n
    int i;
    for(i = 0; i < n; i++)
    {
        if(in[i] == pre[0])  //找到根结点在中序的位置
        {
            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;
    char pre[n+1],in[n+1];  
    cin >> pre >> in;
    cout << dfs(pre,in,n) << endl;
    return 0;
}