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;
}