题目
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC结尾无空行
输出样例:
5
结尾无空行
思路
实现两个函数,第一个是根据先序遍历和中序遍历创建二叉树,第二个是分别求二叉树的左子树和和右子树高度进行比较得出树的高度
根据先序遍历的特点,先序遍历序列中的第一个结点即为根节点;
再从中序遍历的序列中找到根结点的位置,根节点的左半部分就是左子树结点,结点数为index个;右半部分就是右子树结点,结点数为N-index-1个;
再次调用函数时,在中序遍历序列中,左子树结点位置从0开始,右子树结点位置从inorder+index+1开始;在先序遍历序列中,左子树结点位置从preorder+1开始,右子树位置从preorder+index+1开始。
注意递归的退出条件
代码
#include <stdio.h>
#include <stdlib.h>
typedef struct TNode *Tree;
struct TNode
{
char data;
struct TNode *Left;
struct TNode *Right;
};
Tree Build(char preorder[], char inorder[], int n)
{
if(!n) return NULL;
Tree t = (Tree) malloc(sizeof(struct TNode));
t->data = preorder[0];
t->Left = t->Right = NULL;
int index = 0;
for(; index<n; index++)
{
if(inorder[index] == preorder[0]) break;
}
t->Left = Build(preorder+1, inorder, index);
t->Right = Build(preorder+index+1, inorder+index+1, n-index-1);
return t;
}
int Height(Tree T)
{
if(T)
{
int m = Height(T->Left);
int n = Height(T->Right);
if(m < n) return n+1;
else return m+1;
}
return 0;
}
int main()
{
int n;
scanf("%d", &n);
getchar();
char preorder[100];
char inorder[100];
for(int i=0; i<n; i++)
scanf("%c", &preorder[i]);
getchar();
for(int i=0; i<n; i++)
scanf("%c", &inorder[i]);
Tree t = Build(preorder, inorder, n);
int h = Height(t);
printf("%d", h);
return 0;
}