PAT 甲级 1107 Social Clusters

题意：每个人可能有多种爱好，把有相同爱好的人归为一类人，认为他们属于一个social cluster，求有多少个这样的集群，以及每个集群的人数

思路：用一个v数组保存喜欢该爱好的人，然后把该爱好的人全部并起来，形成一个集群，数组的元素的如果是负数且访问过了那么绝对值代表集群的人数

代码：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int root[1010],cnt=0;
vector<int> v[1010],num;
bool vis[1010] = { false };
int findRoot(int s) {
	while (root[s] >= 0)
		s = root[s];
	return s;
}
void Union(int a,int b) {//根据规模大小来进行并，防止树一直长高
	int id1 = findRoot(a);
	int id2 = findRoot(b);
	if (id1 > id2) {
		root[id2] += root[id1];
		root[id1] = id2;
	}else if(id1 < id2){
		root[id1] += root[id2];
		root[id2] = id1;
	}
}
bool com(int a,int b) {
	return a>b;
}
int main() {
	fill(root, root + 1010,-1);
	int n = 0, k = 0,id = 0;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		vis[i] = true;
		scanf("%d:",&k);
		for (int j = 0; j < k; j++) {
			scanf("%d",&id);
			v[id].push_back(i);
		}
	}
	for (int i = 1; i < 1010; i++) {
		for (int j = 0;v[i].size()!=0 && j < v[i].size()-1; j++) {
			Union(v[i][j],v[i][j+1]);
		}
	}
	for (int i = 1; i <= 1010; i++) {
		if (vis[i] == true && root[i] < 0)  {
			cnt++;
			num.push_back(0 - root[i]);
		}
	}
	sort(num.begin(), num.end(), com);
	printf("%d\n",cnt);
	for (int i = 0; i < num.size(); i++) {
		printf("%d%s", num[i], i == num.size() - 1?"\n":" ");
	}
	system("pause");
	return 0;
}