4-12 求二叉树高度 (10分)

4-12 求二叉树高度 (10分)

本题要求给定二叉树的高度。

函数接口定义：

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

思路：

和4-11一样是用递归的思想（别忘了空树的情况！！）

AC代码：

int GetHeight( BinTree BT )
{
    if(BT == NULL)		//空树返回0
        return 0;
    if(BT->Left==NULL&&BT->Right==NULL)		//没有左右孩子，高度为1
        return 1;
    if(BT->Left!=NULL&&BT->Right==NULL)		//右孩子为空，继续计算左子树长度
        return 1+GetHeight(BT->Left);
    if(BT->Left==NULL&&BT->Right!=NULL)		//左孩子为空，继续计算右子树长度
        return 1+GetHeight(BT->Right);
    int a = GetHeight(BT->Left);
    int b = GetHeight(BT->Right);
    if(a >= b)			//如果左右孩子都存在，比较左右子树长度，取长度更大的子树
    	return 1+a;
    else
    	return 1+b;
}

改进后的代码：

int GetHeight( BinTree BT ){
	if(BT == NULL)
	    return 0;
	int left = GetHeight(BT->Left);
	int right = GetHeight(BT->Right);
    if(left >= right)
           return left+1;
    else
           return right + 1;
}