PTA 还原二叉树

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

输入格式:

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

输出格式:

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

输入样例:

9

ABDFGHIEC
FDHGIBEAC

输出样例:

5

程序代码：

#include <iostream>
using namespace std;
// 递归还原二叉树，计算二叉树高度
int dfs(char a[],char b[], int n){
	int i;
	// 节点数为0，高度为0
	if(!n)
	return 0;
	// 在中序中找先序的第一个数
	for( i = 0;i < n;i++){
		if(b[i] == a[0])
		break;
	}
	// 递归，计算左右子树高度
	int l = dfs(a+1 , b , i) + 1; // 先序从第二个开始后i个是左子树
	int r = dfs(a+1+i , b+1+i , n-1-i) + 1; // 先序从第i+1个开始后n-1-i个是右子树
	// 返回二叉树高度
	return l > r ? l : r;
}
int main(){
	int n;
	char a[101],b[101];
	cin >> n >> a >> b;
	cout << dfs(a,b,n);
}