天勤模拟赛1 二叉树问题

题目描述

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

输入格式

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

输出

对于每组输入，输出一个整数，即该二叉树的高度。

样例输入

9
ABDFGHIEC
FDHGIBEAC
7
Abcdefg
gfedcbA

样例输出

5
7

#include <stdio.h>
#include <string.h>

char pre[60], in[60];

typedef struct Tree
{
	Tree *lchild;
	Tree *rchild;
	char data;
};

Tree tree[60];

int loc = 0, n, max;

Tree *creat()
{
	tree[loc].lchild = tree[loc].rchild = NULL;

	return &tree[loc++];
}

Tree *build(int p1, int p2, int i1, int i2)
{
	int i;

	Tree *T = creat();

	T->data = pre[p1];

	for(i = i1; i <= i2; i++)
	{
		if(in[i] == pre[p1])
			break;
	}

	if(i != i1)
		T->lchild = build(p1+1, i - i1 + p1 , i1, i-1);

	if(i != i2)
		T->rchild = build(p2 + 1 + i - i2, p2, i + 1, i2);

	return T;
}

void preOrder(Tree *T, int level)
{
	if(T == NULL)
	{
		if(level > max)
			max = level;
		return;
	}

	preOrder(T->lchild, level + 1);
	preOrder(T->rchild, level + 1);
}
int main()
{
	while(scanf("%d", &n) != EOF)
	{
		Tree *T = NULL;

		scanf("%s", pre);
		scanf("%s", in);

		T = build(0, n-1, 0, n-1);

		max = -1;

		preOrder(T, 0);

		printf("%d\n", max);
		
	}
	return 0;
}