L3-3. 社交集群 PAT团体设计天梯赛

题目 https://www.patest.cn/contests/gplt/L3-3

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
vector<int>a[1002];
int vis[1002],visit[1002];
int num=1,n;
queue<int> q;
vector<int >ans;

int find_r(int hh,int temp)
{
    for(int i=1;i<=n;i++){
        for(int j=0;j<a[i].size();j++){
            if(a[i][j]==hh&&i!=temp){
                num++;
                q.push(i);
                visit[i]=1;
            }
        }
    }
}

void bfs(int pp)
{
    q.push(pp);
    while(!q.empty()){
        int temp=q.front();
        q.pop();
        for(int i=0;i<a[temp].size();i++){
            int hh=a[temp][i];
            if(vis[hh]==0){
                find_r(hh,temp);
                vis[hh]=1;
            }
        }
    }
}

int com(int a,int b)
{
    return a>b;
}

int main()
{
    int k,t;
    cin>>n;
    memset(vis,0,sizeof(vis));
    memset(visit,0,sizeof(visit));
    for(int i=1;i<=n;i++){
        cin>>k;
        getchar();
        while(k--){
            cin>>t;
            a[i].push_back(t);
        }
    }
    for(int i=1;i<=n;i++){
        if(visit[i]==0){
            bfs(i);
           ans.push_back(num);
            num=1;
        }
    }
    sort(ans.begin(),ans.end(),com);
    printf("%d\n",ans.size());
    printf("%d",ans[0]);
    for(int i=1;i<ans.size();i++){
        printf(" %d",ans[i]);
    }

    return 0;
}