任务描述
设二叉树中每个结点的元素均为一个字符,按先序遍历的顺序建立二叉链表,编写三个递归算法对二叉树的结点(度为0、1、2)个数进行统计。
编程要求
输入
多组数据。每组数据一行,为二叉树的前序序列(序列中元素为‘0’时,表示该结点为空)。当输入只有一个“0”时,输入结束。
输出
每组数据输出一行,每行三个数分别为二叉树的度为0、1、2的结点个数。每两个数用空格分隔。
测试说明
测试输入:
abcd00e00f00ig00h00
abd00e00cf00g00
0
预期输出:
5 0 4
4 0 3
#include<iostream>
#include<string.h>
using namespace std;
int a,b,c;//a、b、c分别表示度为0、1、2的结点个数
typedef struct BiTNode
{
char data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
void CreateBiTree(BiTree &T,char S[],int &i)
{//先序建立二叉树
if(S[i]=='0')
T=NULL;
else
{
T=new BiTNode;
T->data=S[i];
CreateBiTree(T->lchild,S,++i);
CreateBiTree(T->rchild,S,++i);
}
}
void Count(BiTree T)
{//二叉树结点个数的统计
if(T==NULL) return ;
if(T->lchild!=NULL&&T->rchild!=NULL) c++;
else if(T->lchild!=NULL||T->rchild!=NULL) b++;
else a++;
Count(T->lchild);
Count(T->rchild);
}
int main()
{
char S[100];
while(cin>>S)
{
if(strcmp(S,"0")==0) break;
a=b=c=0;
int i=-1;
BiTree T;
CreateBiTree(T,S,++i);
Count(T);
cout<<a<<" "<<b<<" "<<c<<endl;
}
return 0;
}