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

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

输入格式:
输入首先给出正整数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;
}