7-23 还原二叉树（求树高）

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

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

代码如下：

/* 思路如下:
1. 根据先序遍历序列第一个结点确定根结点
2. 根据根结点在中序遍历序列中分割出左右两个子序列
3. 对左子树和右子树分别递归使用相同的方法继续分解
*/
#include<stdio.h>
int tree_height(char x[],char z[],int n){
	if (n==0) return 0;
	int i;
	for (i=0;i<n;i++){
		if (x[0]==z[i])
		break;
	}
	int left=tree_height(x+1,z,i)+1;	//左子树
	int right=tree_height(x+i+1,z+i+1,n-i-1)+1;	//右子树
	return left>right?left:right;
}
int main()
{
	int n;
	scanf("%d",&n);
	getchar();
	char x[n];
	char z[n];
	scanf("%s%s",x,z);
	int height=tree_height(x,z,n);
	printf("%d",height);
	return 0;
}