本题要求给定二叉树的高度。
函数接口定义:
int GetHeight( BinTree BT );
其中BinTree
结构定义如下:
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
ElementType Data;
BinTree Left;
BinTree Right;
};
要求函数返回给定二叉树BT的高度值。
裁判测试程序样例:
#include <stdio.h>
#include <stdlib.h>
typedef char ElementType;
typedef struct TNode *Position;
typedef Position BinTree;
struct TNode{
ElementType Data;
BinTree Left;
BinTree Right;
};
BinTree CreatBinTree(); /* 实现细节忽略 */
int GetHeight( BinTree BT );
int main()
{
BinTree BT = CreatBinTree();
printf("%d\n", GetHeight(BT));
return 0;
}
/* 你的代码将被嵌在这里 */
输出样例(对于图中给出的树):
4
思路:如果是空树,则高度为0。否则,递归计算左子树高度为m,右子树高度为n,则二叉树高度为m,n中较大的数再加1
代码:
int GetHeight( BinTree BT )
{
int m, n;
if (BT == NULL) //空树返回0
return 0;
m = GetHeight(BT->Left); //递归求左子树高度
n = GetHeight(BT->Right); //递归求右子树高度
if (m > n)
return m + 1;
else
return n + 1;
}
一行代码精简版:
int GetHeight( BinTree BT )
{
return (BT ? (GetHeight(BT->Left) > GetHeight(BT->Right) ? GetHeight(BT->Left) + 1 : GetHeight(BT->Right) + 1) : 0);
}