7-23 还原二叉树 (25分)

tech2023-11-11  76

给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。

输入格式: 输入首先给出正整数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; }
最新回复(0)