17121 求二叉树各种节点数

SCAU数据结构OJ第四章

17121 求二叉树各种节点数

Description
构造二叉链表表示的二叉树：按先序次序输入二叉树中结点的值（一个字符），'#'字符表示空树，
构造二叉链表表示的二叉树T,并求此二叉树中度为2的节点总数，度为1的节点总数，度为0的节点总数

输入格式
第一行输入先序次序二叉树中结点

输出格式
第一行输出度为2的节点数
第二行输出度为1的节点数
第三行输出度为0的节点数

输入样例
ABC###D##

输出样例
1
1
2

代码如下：

#include "stdio.h"
#include "malloc.h"
#define TRUE 1
#define FALSE 0
#define OK  1
#define ERROR  0
#define INFEASIBLE -1
#define OVERFLOW -2
typedef int  Status;
typedef char  ElemType;
typedef struct BiTNode
{
    ElemType data;
    struct BiTNode *lchild,*rchild;//左右孩子指针
} BiTNode,*BiTree;
int num0 = 0,num1 = 0,num2 = 0;//分别代表度为0，1，2的数量

Status CreateBiTree(BiTree &T)    // 算法6.4
{
    // 按先序次序输入二叉树中结点的值（一个字符），’#’字符表示空树，
    // 构造二叉链表表示的二叉树T。
    char ch;
    scanf("%c",&ch);
    if (ch=='#')
        T = NULL;
    else
    {
        if (!(T = (BiTNode *)malloc(sizeof(BiTNode))))
            return ERROR;
        T->data = ch;// 生成根结点
        CreateBiTree(T->lchild);   // 构造左子树
        CreateBiTree(T->rchild);  // 构造右子树

        if(T->lchild&&T->rchild)
            num2++;
        else if((!T->lchild&&T->rchild)|| (T->lchild&&!T->rchild))
            num1++;
        else if(!T->lchild&&!T->rchild)
            num0++;
    }
    return OK;
} // CreateBiTree

int main()   //主函数
{
    BiTree T;
    CreateBiTree(T);                    //补充代码
    printf("%d\n%d\n%d\n",num2,num1,num0);
    return 0;
}//main