二叉树高度

题目描述

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

 

输入

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

 

输出

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

 

样例输入

9
ABDFGHIEC
FDHGIBEAC

样例输出

5

参考代码：

#include<bits/stdc++.h>
using namespace std;
string s1,s2;
int len,n;
int find(char x){
    for(int i=0;i<=len;i++)
        if(s2[i]==x)
            return i;
}
int pre(int l1,int r1,int l2,int r2){
    int ip=find(s1[l1]),k1=0,k2=0;
    if(ip>l2){
        k1+=pre(l1+1,l1+ip-l2,l2,ip-1)+1;
    }
    if(ip<r2){
        k2+=pre(ip-l2+l1+1,r1,ip+1,r2)+1;
    }
    return max(k1,k2);
}
int main(){
    cin>>n>>s1>>s2;
    len=s1.size()-1;
    cout<<pre(0,len,0,len)+1;
    return 0;
}