给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
结尾无空行
输出样例:
5
结尾无空行
AC代码:
#include<iostream>
#include<string>
using namespace std;
int Height(string P, string In) {
if (P==""|| In == "")return 0;//出口条件,一定记得设置
int root = P[0]; int pos = In.find(root);
string P1 = P.substr(1,pos);
string In1 = In.substr(0, pos);
string P2 = P.substr(pos + 1, P.size()-1-pos);
string In2 = In.substr(pos + 1, P.size() - 1 - pos);
int lh = Height(P1, In1);
int rh = Height(P2, In2);
return 1 + (lh > rh ? lh : rh);
}
int main() {
int N; cin >> N;
string pre, in;
cin >> pre >> in;
cout << Height(pre, in);
}
注意点:
1、对于递归函数的设计一定要记得设置递归的最边界情况的出口
2、substr() 函数用于从 string 字符串中提取子字符串,它的原型为:
string substr (size_t pos = 0, size_t len = npos) const;
pos 为要提取的子字符串的起始下标,len 为要提取的子字符串的长度。