每日一题Day55

基于二叉链表的二叉树结点个数的统计

描述

设二叉树中每个结点的元素均为一个字符，按先序遍历的顺序建立二叉链表，编写三个递归算法对二叉树的结点（度为0、1、2）个数进行统计。

输入

多组数据。每组数据一行，为二叉树的前序序列（序列中元素为‘0’时，表示该结点为空）。当输入只有一个“0”时，输入结束。

输出

每组数据输出一行，每行三个数分别为二叉树的度为0、1、2的结点个数。每两个数用空格分隔。

样例输入1 

abcd00e00f00ig00h00
abd00e00cf00g00
0

样例输出1

5 0 4
4 0 3

解答：根据输入字符串建立二叉链表。在遍历二叉树的同时统计二叉树中度为0、1、2的结点个数。

#include<stdio.h>
#include<stdlib.h>

int k;
int count0,count1,count2;
typedef struct node
{
	char data;
	struct node *left,*right;
} Node,*BiTree;

void Create(BiTree &T,char *s)
{
	k++;
	if(s[k]=='0')
		T=NULL;
	else
	{
		T=(Node *)malloc(sizeof(Node));
		T->data = s[k];
		Create(T->left,s);
		Create(T->right,s);
	}
}

void Count(BiTree T)
{
	if(T)
	{
		if(T->left && T->right)
			count2++;
		else if(T->left || T->right)
			count1++;
		else
			count0++;
		Count(T->left);
		Count(T->right);
	}
}

int main()
{
	BiTree T;
	char s[100];
	while(1)
	{
		scanf("%s",s);
		if(s[0]=='0' && s[1]=='\0')
			break;
		k=-1;
		count0=count1=count2=0;
		Create(T,s);
		Count(T);
		printf("%d %d %d\n",count0,count1,count2);
	}
	return 0;
}