PTA（树和二叉树）1-2 二叉树求结点数

1-2 二叉树求结点数

分数 3

作者 鲁法明

单位 山东科技大学

编写函数计算二叉树中的节点个数。二叉树采用二叉链表存储结构。

函数接口定义：

1
int NodeCountOfBiTree ( BiTree T);

其中 T是二叉树根节点的地址。

裁判测试程序样例：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
//头文件包含
#include<stdlib.h>
#include<stdio.h>
#include<malloc.h>

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

typedef int Status;

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

//创建二叉树各结点，输入0代表创建空树。
//采用递归的思想创建
//递归边界：空树如何创建呢：直接输入0；
//递归关系：非空树的创建问题，可以归结为先创建根节点，输入其数据域值；再创建左子树；最后创建右子树。左右子树递归即可完成创建！
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 NodeCountOfBiTree ( BiTree T);
//下面是主函数
int main()
{
   BiTree T;
   int n;     
   CreateBiTree(T);  //先序递归创建二叉树     
   n= NodeCountOfBiTree(T);     
   printf("%d",n);
     return 0;
}

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

输入样例（注意输入0代表空子树）：

1
1 3 0 0 5 3 0 0 0

输出样例：

1
4

我的第一遍答案（错误）

1
2
3
4
5
6
7
int NodeCountOfBiTree (BiTree T) {
    // 空树，只有自己一个根节点，返回1
    if (T == NULL) {
        return 1;
    }
    return 1 + NodeCountOfBiTree(T->lchild) + NodeCountOfBiTree(T->rchild);
}
  • 发现，空树是一个结点都没有，指针所指向的是空，应该return 0；

我的正确答案

1
2
3
4
5
6
7
8
int NodeCountOfBiTree (BiTree T) {
    // 空树，只有自己一个根节点，返回0
    if (T == NULL) {
        return 0;
    } else {
        return 1 + NodeCountOfBiTree(T->lchild) + NodeCountOfBiTree(T->rchild);
    }
}