天梯赛练习题 L3-003 社交集群 dfs

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：K​i​​ : h​i​​ [1] h​i​​ [2] … h​i​​ [K​i​​ ]其中K​i​​ (>0)是兴趣爱好的个数，h​i​​ [j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

输出样例：

3
4 3 1

分     析：题目要求计算拥有共同爱好的群体的个数以及各个群体的人的数量。首先就是想到构造一个图，然后深度优先遍历节点，每次得到一个团体的数量，最后遍历完成即可得到答案。
     那么如何构造图？也就是怎样决定两个人是否相连？将每个人的兴趣读入一个set，然后查看两个人的兴趣是否有交集从而决定两个人是否相连？感觉不是很elegant（会不会超时就更不知道了）。
    在本题中，人都是通过兴趣联系起来，所以可以将兴趣看作假人（计算群体人数时不算入）。由于人的N$\le$ 1000，所以人的编号的可能取值处于[1,1000]，兴趣的编号也是[1,1000]内的整数，所以直接将兴趣当作人存入肯定是不行的，所以这里将每个兴趣的编号都加1000（也就是原来编号为1的兴趣变为1001），这样dfs的时候只需查看当前节点的id是否大于1000，若大于则是假人，不计人数，若小于1000则说明是真人，人数加1。

代码如下：

#include <bits/stdc++.h>
using namespace std;

int G[2005][2005];
bool visited[1005];

bool cmp(int& a,int& b)
{
	return a > b;
}

int dfs(int u) 
{	
	visited[u] = true;
	int sum = 0;                // u号节点（可能是兴趣，可能是人）拉进来的人 
	for(int i = 1;i < 2005; i++)
	{
		if(i <= 1000 && !visited[i] && G[u][i] == 1)    // 如果是人 
			sum += 1+dfs(i);                            // 自身加他拉进来的人 
		else if(i > 1000 && G[u][i] == 1)               // 如果是兴趣 
			sum += dfs(i);
	}
	return sum;
}

int main(int argc, char const *argv[])
{
	int n,k,t;
	cin >> n;
	memset(G,0,sizeof(G));
	
	for(int i = 1;i <= n; i++)
	{
		scanf("%d: ",&k);
		for(int j = 0;j < k; j++)
		{
			cin >> t;
			G[i][1000+t] = G[1000+t][i] = 1;
		}
	}
	
	vector<int> ans;
	for(int i = 1;i <= n; i++)
	{
		if(!visited[i])
			ans.push_back(dfs(i)+1);    // 1表示自己，dfs(i)表示和i号有相同兴趣的人的数量 
	}
	sort(ans.begin(),ans.end(),cmp);
	cout << ans.size() << endl;
	for(int i = 0;i < ans.size(); i++)
	{
		cout << ans[i];
		if(i != ans.size()-1)
			cout << " ";
	} 
	cout << endl;
	return 0;
}