7-23 还原二叉树（25 分）

版权声明：Dirichlet_zju https://blog.csdn.net/Dirichlet_zju/article/details/83722099

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

• 分析：先序遍历首先遍历根节点，中序遍历首先遍历左子树。因此可以根据先序遍历第一个点将中序遍历分成左右子树。注解均在其中–>

#include <iostream>
//还原二叉树 
//给出先序遍历和中序遍历，还原出树，给出输的高度 
using namespace std;

int GetHeight(char* Pre, char* Ino, int N){//参数包括：先序遍历，中序遍历，节点个数 
	int Root;
	int H1, H2;//左右子树高度 
	
	if(N==0) return 0;//当遍历到结束，N=0，此即递归初始条件！ 
	for(int i=0; i<N; i++){//找到根节点位置 
		if(Pre[0]==Ino[i]){
			Root=i;
			break;
		}
	}
	
	/*做右子树时注意，中序遍历Root所指向点的后面全都是右子树，且先序遍历右子树与其等长*/ 
	H1=GetHeight(Pre+1, Ino, Root)+1;//左子树+1 
	H2=GetHeight(Pre+Root+1, Ino+Root+1, N-Root-1)+1;//右子树+1 
	
	return H1>H2?H1:H2;//两者比较返回较大值
} 

int main(){
	int n;
	char pre[101];
	char ino[101];
	 
	cin>>n>>pre>>ino;
	
	cout<<GetHeight(pre, ino, n)<<endl;
	return 0;
}