7-23 还原二叉树 (25分)
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
思路:根据前序遍历和中序遍历来确定左右子树,先找到树根即前序遍历第一位,中序序列的树根前面的字符串是其左子树,后面是右子树。再将左子树和右子树的序列分别作为一棵树进入递归,对其重复找出树根和左右子树的处理,每次记录下当前左右子树的深度,最后返回最大深度。
#include<iostream>
using namespace std;
int dfs(string a,string b,int num){
int i;
if(num==0) return 0;
for(i=0;i<num;i++)
{
if(b[i]==a[0])
break;
}
int x=dfs(a.substr(1),b,i)+1;//求左子树的深度
int y=dfs(a.substr(i+1),b.substr(i+1),num-i-1)+1;//求右子树的深度
return x>y?x:y;
}
int main()
{
string pre_order,mid_order;
int num;
cin>>num;
cin>>pre_order>>mid_order;
int cnt=dfs(pre_order,mid_order,num);
cout<<cnt<<endl;
}