一、题目概述
依据给定N个用户的爱好,构造出一个社交网络,找到社交网络中的同好圈的个数,并降序输出各同好圈中的成员数。
二、思路
1、同好圈的示例:
A的爱好:1, 3;
B的爱好:3, 5;
C的爱好:5, 6;
显然:A、B是关于3的同好,B、C是关于5的同好。A、B、C组成了一个人同好圈,即使A、C没有共同爱好。
2、解决方案
构造无向图,统计连通分量数、各连通分量中用户结点数。标号范围[1, 1000]内为爱好结点,标号范围[1001, 1000 + N]为用户结点。dfs遍历无向图进行统计。
三、代码
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
vector<int> G[2001], visited(2001, false), ans;
int N, cnt;
void dfs( int v )
{
++visited[v];
if( v > 1000 )
++cnt;
for( int i = 0; i < G[v].size(); ++i )
if( !visited[G[v][i]] )
dfs(G[v][i]);
}
int main()
{
scanf("%d", &N);
for( int i = 1, K; i <= N; ++i )
{
scanf("%d:", &K);
for( int j = 0, id; j < K; ++j )
{
scanf("%d", &id);
G[id].push_back(1000 + i);
G[1000 + i].push_back(id);
}
}
for( int i = 1; i <= N; ++i )
if( !visited[1000 + i] )
{
cnt = 0;
dfs(1000 + i);
ans.push_back(cnt);
}
printf("%d\n", ans.size());
sort(ans.begin(), ans.end(), [](int a, int b){return a > b;});
for( int i = 0; i < ans.size(); ++i )
printf("%s%d", i ? " ":"", ans[i]);
}