给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
思路:只要中序的string还有数据(即size>0),我们便执行操作。并且记录最大的一个高度(高度从1开始记录)。最后输出我们求出的答案。
#include<iostream>
#include<cmath>
using namespace std;
int ans = 0;
void LXR(string xx, string zx,int d) {
if (zx.size() > 0) {
int f = zx.find(xx[0]);
ans = max(ans, d);
LXR(xx.substr(1, f), zx.substr(0, f), d + 1);
LXR(xx.substr(f + 1), zx.substr(f + 1), d + 1);
}
}
int main() {
int n;
cin >> n;
string xx, zx;
cin >> xx >> zx;
LXR(xx, zx, 1);
cout << ans;
return 0;
}