二叉树问题

问题 D: 二叉树问题

时间限制: 1 Sec  内存限制: 32 MB
提交: 208  解决: 96
[提交][状态][讨论版]

题目描述

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

输入

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

输出

对于每组输入，输出一个整数，即该二叉树的高度。

样例输入

9
ABDFGHIEC
FDHGIBEAC
7
Abcdefg
gfedcbA

样例输出

5
7

#include<iostream>
#include<string>
using namespace std;
char s1[60],s2[60];
int dfs(char a[],char b[],int m)
{
    int i;
    if(m==0) return 0;//没有节点，为空树
    for(i=0;i<m;i++)
    {
        if(b[i]==a[0]) break;//找到根节点在中序的位置  
    }
    int c=dfs(a+1,b,i)+1;//左子树的深度
    int d=dfs(a+i+1,b+i+1,m-i-1)+1;//右子树的深度
    return c>d?c:d;
}
int main()
{
    int n;
    while( cin>>n){
    cin>>s1>>s2;
    cout<<dfs(s1,s2,n)<<endl;
}
    return 0;
 
}