DFS(连通分量) - 社交集群 - 天梯赛L3-003

DFS(连通分量) - 社交集群 - L3-003

分析：

画出样例的图可以发现，本质上就是求一个连通块内，特殊点的个数。

我们将兴趣爱好编号统一增加1000，与人的编号分离出来，然后按照题意建图。

接着遍历整个图，统计连通块的个数，以及每个连通块内，人的个数。

最后排序，输出。

点的个数 ≤1000+1000 = 2000，时间复杂度O(n)

代码：

#include<sstream>
#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 2010, M = 1000010;

int e[M], ne[M], h[N], idx;
int n, p[N], cnt, s;
bool st[N];

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void dfs(int u)
{
    if(u <= 1000) s ++;
    st[u] = true;
    for(int i=h[u];~i;i=ne[i])
    {
        int j = e[i];
        if(!st[j]) dfs(j);
    }
}

int main()
{
    scanf("%d", &n);
    int k, x;
    memset(h, -1, sizeof h);
    for(int i=1;i<=n;i++)
    {
        scanf("%d:", &k);
        while(k --)
        {
            scanf("%d", &x);
            add(x + 1000, i), add(i, x + 1000);
        }
    }
    
    for(int i=1;i<=n;i++)
        if(!st[i])
        {
            s = 0;
            dfs(i);
            p[cnt ++] = s;
        }
    
    sort(p, p + cnt, greater<int>());
    
    printf("%d\n", cnt);
    for(int i=0;i<cnt;i++) printf("%d%s", p[i], i == cnt-1 ? "" : " ");
    
    return 0;       
}