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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

思路:

通过先序序列找出根节点,通过中序序列的【特性】找出左右子树

中序序列的特性:在根节点左侧的序列是左子树,右侧是右子树;

通过根节点,左右子树递归的生成树,最终算出所生成树的高度;

一开始打算用中序序列的两端端点来做,然而在递归出口那里一直有问题,于是上博客找各路大神的解法,他们直接把序列分成两半,递归出口就是判断序列是否为空,完美解决了。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct tree{
	char data;
	struct tree *left,*right;
}Tree;
//先序第一个确定根节点,中序在根节点左边全是左子树,右边全是右子树 
Tree *creat(char pre[],char mid[],int n)//分别传入先序序列,中序序列,剩余元素的个数
{
	if(n==0)return NULL;
	Tree *t=(Tree*)malloc(sizeof(Tree)); 
	int index=0;
	while(index<n)
	{
		if(mid[index]==pre[0])break;
		index++;
	}
	
	t->data=pre[0];//根节点
	t->left=creat(pre+1,mid,index);//左边确定左子树 
	t->right=creat(pre+index+1,mid+index+1,n-index-1); //右边确定右子树
	return t; 
 } 
int getH(Tree *T)
{
	if(!T)return 0;
	
	int lefth=getH(T->left);
	int righth=getH(T->right);
	
	return lefth>righth?++lefth:++righth;
}
int main()
{	
	int N,HIGH;
	char pre[55],mid[55];
	scanf("%d",&N);
	getchar();
	gets(pre);
	gets(mid);
	Tree *T=creat(pre,mid,N);
	HIGH=getH(T); 
	printf("%d",HIGH);
	return 0;
}

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐