二叉树求深度和叶子数(20 分)

编写函数计算二叉树的深度以及叶子节点数。二叉树采用二叉链表存储结构

函数接口定义:

int GetDepthOfBiTree ( BiTree T);
int LeafCount(BiTree T);

其中 T是用户传入的参数,表示二叉树根节点的地址。函数须返回二叉树的深度(也称为高度)。

裁判测试程序样例:


//头文件包含
#include
#include
#include

//函数状态码定义
#define TRUE       1
#define FALSE      0
#define OK         1
#define ERROR      0
#define OVERFLOW   -1
#define INFEASIBLE -2
#define NULL  0
typedef int Status;

//二叉链表存储结构定义
typedef int TElemType;
typedef struct BiTNode{
    TElemType data;
    struct BiTNode  *lchild, *rchild; 
} BiTNode, *BiTree;

//先序创建二叉树各结点
Status CreateBiTree(BiTree &T){
   TElemType e;
   scanf("%d",&e);
   if(e==0)T=NULL;
   else{
     T=(BiTree)malloc(sizeof(BiTNode));
     if(!T)exit(OVERFLOW);
     T->data=e;
     CreateBiTree(T->lchild);
     CreateBiTree(T->rchild);
   }
   return OK;  
}

//下面是需要实现的函数的声明
int GetDepthOfBiTree ( BiTree T);
int LeafCount(BiTree T);
//下面是主函数
int main()
{
   BiTree T;
   int depth, numberOfLeaves;
   CreateBiTree(T);
   depth= GetDepthOfBiTree(T);
	 numberOfLeaves=LeafCount(T);
   printf("%d %d\n",depth,numberOfLeaves);
}

/* 请在这里填写答案 */

输入样例:

1 3 0 0 5 7 0 0 0

输出样例:

3 2
int GetDepthOfBiTree(BiTree T)
{
    int d1 = 0, d2 = 0, d;
    if(!T)
        return 0;
    else
    {
       d1 = GetDepthOfBiTree(T->lchild) + 1;
       d2 = GetDepthOfBiTree(T->rchild) + 1;
    }
    d = d1 > d2 ? d1 : d2;
    return d;
}
int LeafCount(BiTree T)
{
    int n = 0;
    if(!T)
        return 0;
    else
    {
        if(T->lchild == NULL && T->rchild == NULL)
           n++;
        else
        {
            n = LeafCount(T->lchild) + LeafCount(T->lchild);
        }
    }
    return n;
}

你可能感兴趣的:(数据结构-PTA)