7-1 还原二叉树

分数 25

作者 DS课程组

单位 浙江大学

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

#include<bits/stdc++.h>
using namespace std;
int dfs(char a[],char b[],int n)
{
    if(n<=0)return 0;
    int i;
    for(i=0;i<n;i++) if(a[0]==b[i])break;
    int x=dfs(a+1,b,i)+1;
    int y=dfs(a+i+1,b+i+1,n-i-1)+1;
    return x>y?x:y;
}
int main()
{
    char a[51],b[51];
    int n;
    cin>>n>>a>>b;
    cout<<dfs(a,b,n);
}