题目描述
现给定一棵二叉树的先序遍历序列和中序遍历序列,要求你计算该二叉树的高度。
输入
输入包含多组测试数据,每组输入首先给出正整数N(<=50),为树中结点总数。下面2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出
对于每组输入,输出一个整数,即该二叉树的高度。
样例输入
9
ABDFGHIEC
FDHGIBEAC
7
Abcdefg
gfedcbA
样例输出
5
7
思路见代码
#include<bits/stdc++.h>
using namespace std;
int dfs(char*pre,char*in,int n)//n为中序遍历中的查找个数,即为子树的节点数
{
if(n==0) return 0;
int i;
for(i=0;i<n;i++) //找顶点
if(pre[0]==in[i])
break;
int left=dfs(pre+1,in,i);//对左子树
//顶点在先序遍历中的位置往后一位
//中序查找的起点不变,但结尾处应小于i ,即n=i
int right=dfs(pre+i+1,in+i+1,n-i-1);// 对右子树
//顶点在先序遍历中的位置要加上其左子树的全部节点数i,再往后移一位
//中序查找的位置就从i往后一位, 其个数为n-i-1;
return max(left,right)+1;//要最深的那个
}
int main()
{
int n;
while(cin>>n)
{
char pre[n+1],in[n+1];
cin>>pre>>in;
cout<<dfs(pre,in,n)<<endl;
}
return 0;
}