浙大数据结构期末：7-1 还原二叉树 (8 分)

7-1 还原二叉树 (8 分)

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

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

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

输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5

代码

#include<iostream>
#include <string>
using namespace std;

int count(string s1, string s2, int N) {
	if (s1.size()==1)return 1;
	else {
		int res = 0;
		int Pos;
		string ls1, ls2,rs1,rs2;
		char ch = s1[0];
		Pos = s2.find(ch);
		ls1 = s1.substr(1, Pos );
		ls2 = s2.substr(0, Pos );
		rs1 = s1.substr(Pos + 1, N - Pos);
		rs2 = s2.substr(Pos + 1, N - Pos);
		int LH, RH;
		LH = count(ls1, ls2, Pos);
		RH = count(rs1, rs2, N - Pos);
		res= (RH > LH ?  RH : LH) + 1;
		return res;
	}
}


int main() {
	string pre, mid;
	int N;
	cin >> N;
	cin >> pre >> mid;
	cout<<count(pre, mid, N);

	return 0;
}

测试结果

总结

总体来说和求二叉树的高度那题差不多，不过稍微变化了一些，但是根本思路是一样的。

这里使用了string的一些特性，比如

string.find(char ch)，会返回ch字符的下标;

还有

string.substr(char* begin_index,int  NumToCopy);