给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式: 输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式: 输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
#include <iostream>
#include <algorithm>
using namespace std
;
int dfs(char* pre
, char* in
, int n
) {
if (n
== 0) return 0;
int i
= n
;
while (--i
)
if (pre
[0] == in
[i
])
break;
return max(dfs(pre
+ 1, in
, i
) + 1, dfs(pre
+ i
+ 1, in
+ i
+ 1, n
- i
- 1) + 1);
}
int main() {
int N
;
char pre
[51], in
[51];
cin
>> N
>> pre
>> in
;
cout
<< dfs(pre
, in
, N
);
return 0;
}