【个人做题笔记】1004 Counting Leaves (30 分)

一开始没想到这么简单emmmmmm
就直接开个结构体，存有无孩子，父节点，以及所在层数。
然后把数据读进来，更新每个节点的父节点。
然后n方暴力跑一次，找到每个节点的子节点，让子节点的层数比父节点大1，,然后基本就结束了，统计每层的叶子结点个数，并记录最大层数用于后面输出答案。
后面其实也可以用dfs跑一遍的，因为有了每个节点与其父、子节点的关系，只要把结构体里有无孩子改成一个用于存子节点序号的vector就行。
代码

//1004 Counting Leaves (30 分)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
#define FOR(i,a,n) for (int i=a;i<n;i++)
#define FRO(i,n,a) for (int i=n;i>a;i--)
#define mst(a,i) memset(a,i,sizeof a)
#define pb push_back 
#define mp make_pair
#define fi first
#define se second
#define pi pair<int,int> 
#define eps 1e-6 
#define PI acos(-1.0)
#define INF 0x3f3f3f3f 
const int mod = 1000000007;
int dir[4][2]={0,1,1,0,0,-1,-1,0};
const int N = 120;
// const int MAXN=1e4;
// int a[N]={};
// int b[N]={};
struct node
{
    int level = -1;
    int fa = -1;
    int child = -1;
};
int sum[101];
vector<node> no(120);
int main(){
    int n , m;
    cin >> n >> m;
    int mx = 1;
    no[1].level = 1;
    FOR(i,0,m)
    {
        int now;
        cin >> now;
        int num;
        cin >> num;
        FOR(j,0,num)
        {
            int k;
            cin >> k;
            no[now].child = 1;
            no[k].fa = now;
        }
    }
    FOR(i,1,n + 1)
    {
        FOR(j,1,n + 1)
        {
            if(no[j].fa == i)
            {
                no[j].level = no[i].level + 1;
                mx = max(no[j].level , mx);
            }
        }
    }
    FOR(i,1,n + 1)
    {
        if(no[i].child == -1)
        {
            sum[no[i].level]++;
        }
        // cout << i << ' ' << no[i].fa << ' ' << no[i].level << '\n';
    }
    FOR(i,1,mx)
    {
        cout << sum[i] << ' ';
    }
    cout << sum[mx] <<'\n';
    return 0;
}