【HBU】4-14 还原二叉树 (15分)

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

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

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

两个知识点，根据前序、中序遍历还原二叉树；求二叉树的高度。

可以参看视频讲解 浙江大学-数据结构

#include <iostream>
using namespace std;
char pre[52], in[52];
typedef struct bitnode{
	char data;
	struct bitnode *lchild, *rchild;
}BiTNode, *BiTree;
BiTree buildTree(int root, int start, int end){
	if(start > end) return NULL;
	BiTree p = (BiTNode *)malloc(sizeof(BiTNode));
	p->data = pre[root];
	int i, cnt = 0;
	for(i = start; i < end; i++){
		cnt++;
		if(in[i] == pre[root])
			break;
	}
	p->lchild = buildTree(root+1, start, i-1);
	p->rchild = buildTree(root+cnt, i+1, end);
	return p;
}
int getHeight(BiTree bt){
	if(bt){
		int l, r;
		l = getHeight(bt->lchild);
		r = getHeight(bt->rchild);
		return max(l, r)+1;
	}
	return 0;
}
int main(){
	int n;
	cin >> n;
	cin >> pre >> in;
	BiTree p = buildTree(0, 0, n-1);
	int hight = getHeight(p);
	cout << hight;
}

附：由后序、中序遍历还原二叉树代码

BiTree setTree(int root, int begin, int end, int post[], int in[]){
	if(begin > end) return NULL;
	BiTree t = (BiTNode*)malloc(sizeof(BiTNode));
	t->data = post[root];
	int i, cnt = 0;
	for(i = begin; i < end; i++){
		cnt++;
		if(in[i] == post[root])
			break;
	}
	t->lchild = setTree(root-1-(end-i), begin, i-1, post, in);
	t->rchild = setTree(root-1, i+1, end, post, in);
	return t;
}