pta 数据结构与算法题目集 7-23 还原二叉树 (25 分)(根据先序中序得后序）

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

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

输入格式:
输入首先给出正整数N（≤50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。

输出格式:
输出为一个整数，即该二叉树的高度。

输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
原题链接
solution：找到前序结点在中序中的位置，则在中序该位置左边都是左子树的结点，右边都是右子树，按照这个思路来建树获取它的高度

#include <bits/stdc++.h>
using namespace std;

char pre[50], in[50];

int dfs(int prel, int prer, int inl, int inr)
{
	if (prel > prer)return 0;
	if (prel == prer)return 1;
	int pos = inl;
	while (in[pos] != pre[prel])++pos;
	int left = dfs(prel + 1, prel + pos - inl, inl, pos - 1) + 1;
	int right = dfs(prel + pos - inl + 1, prer, pos + 1, inr) + 1;
	return max(left, right);
}

int main()
{
	int n;
	cin >> n >> pre >> in;
	cout << dfs(0, n - 1, 0, n - 1) << endl;
	return 0;
}