7-12 二叉树每层度为1节点数 (10 分)
编写程序统计一棵非空二叉树中每层度为1的结点的数目,二叉树结点个数不超过100。
输入格式:
输入为一个字符串,表示带空指针信息的二叉树先根序列。其中空指针信息用#表示,二叉树结点为a-z, A-Z的字母。
输出格式:
输出为若干行,按层数从小到大次序输出二叉树每层度为1的结点个数,即第1行输出第0层度为1的结点个数,第2行输出第1层度为1的结点个数,以此类推。
输入样例:
ABD###CE###
输出样例:
0
2
0
思路很简单,就是建树然后根据层次遍历没层的节点,然后统计每层度为一的节点。
#include<iostream>
#include<queue>
using namespace std;
int a[200]={0};
typedef struct Node{
struct Node *l,*r;
char data;
}*tree,node;
queue<pair<tree,int> > Q; //记录节点和所在层数
int Max=0;
void create(tree &T)
{
char c;
cin>>c;
if(c=='#')
T=NULL;
else
{
T=new node;
T->data=c;
create(T->l);
create(T->r);
}
}
void bfs(tree T)
{
Q.push(make_pair(T,1));
while(!Q.empty())
{
pair<tree,int> p;
p=Q.front();
tree s=p.first;
Max=max(Max,p.second);
Q.pop();
if(s->l&&s->r==NULL)
{ a[p.second]++;
Q.push(make_pair(s->l,p.second+1));
}else if(s->l==NULL&&s->r)
{
a[p.second]++;
Q.push(make_pair(s->r,p.second+1));
}else if(s->l&&s->r)
{
Q.push(make_pair(s->l,p.second+1));
Q.push(make_pair(s->r,p.second+1));
}
}
}
int main()
{
tree T;
create(T);
bfs(T);
for(int i=1;i<=Max;i++)
{
printf("%d\n",a[i]);
}
}
系兄弟就来砍我,不要嫖我。