之前一直WA,转换了个思路(用一个parent[]保存父结点,而不是老想用一个数组保存子结点),就1Y了。输入时进行处理,用一个parent[]保存父结点,标记出所有的非叶结点,再对每个叶结点计算它们的层数,相应的层数的叶结点数++即可。
AC代码:
#include<iostream>
using namespace std;
struct node
{
node *left,*right;
int value;
};
const int NUM=105;
int parent[NUM],level[NUM],isLeaf[NUM];
int maxLevel=1;
void findLevel(int x)
{
int r=x;
int temp=1;//temp是叶结点的层数
while(parent[r]!=r){
temp++;
r=parent[r];
}
level[temp]++;//所在层数的叶结点++
if(temp>=maxLevel)//求出最大层数
maxLevel=temp;
}
int main()
{
int n,m,i,id,k,j;
cin>>n>>m;
for(i=0;i<NUM;i++){
parent[i]=i;
isLeaf[i]=1;
}
for(i=0;i<m;i++){
cin>>id>>k;
while(k--){
cin>>j;
parent[j]=id;
}
isLeaf[id]=0;//将id标记为非叶结点
}
for(i=1;i<=n;i++){
if(isLeaf[i]==1){//如果是叶结点,再进行层数计算
findLevel(i);
}
}
cout<<level[1];
for(i=2;i<=maxLevel;i++){
cout<<" "<<level[i];
}
return 0;
}