7-23 还原二叉树 (25 分)（递归）

参考：借用大神的解法

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int shu(char a[] , char b[] , int n){
    int k;
    if(n == 0) return 0;
    for(int i = 0 ; i < n ; i++){
        if(b[i] == a[0]){
            k = i;
            break;
        }
    }
    int x = shu(a+1 , b , k) + 1;
    int y = shu(a+k+1 , b+k+1 , n-k-1) + 1;
    return x > y ? x : y;
}
int main(){
    char a[55] , b[55];
    int n;
    cin>>n;
    cin>>a>>b;
    int gao = shu(a , b , n);
    cout<<gao;
}