PAT A 1107 Social Clusters (30分)

一、题目概述

依据给定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]);
}