30分的大水题 题意就是给出一颗族谱树(不一定是二叉树)然后让你求出每一层的叶子节点有几个。
直接用邻接矩阵存储树 由于给定了根节点 直接dfs这颗树 在dfs的过程中顺便记录下来非叶子节点的个数,最后输出,贴代码。
#include<bits/stdc++.h>
using namespace std;
int m,n;
int g[300][300];
int cemax = -999;
int ans[300];
int dfs(int fa,int id,int ce)//fa是父节点,为了防止向上一层dfs造成死循环,ce为层数
{
if(ce > cemax) cemax=ce; //记录下最大层数,一会方便输出
int flag = 1;
for(int i=0;i<m+1;i++)
if(g[id][i])
if(i!=fa)
{
dfs(id,i,ce+1);
flag = 0;
}
if(flag) ans[ce] += 1;//如果flag的值没有被修改过说明改节点没有相连的子节点
}
int main()
{
scanf("%d%d",&m,&n);
for(int i=0;i<n;i++)
{
int id,k,temp;
scanf("%d%d",&id,&k);
while(k--)
{
scanf("%d",&temp);
g[id][temp] = g[temp][id] = 1; //邻接矩阵存储树
}
}
dfs(-1,1,0);
for(int i=0;i<=cemax;i++)
{
printf("%d",ans[i]);
if(i!=cemax) printf(" ");
}
printf("\n");
}