1107 Social Clusters (30 分)
题目大意
有n个人,每个人喜欢k个活动,如果两个人有任意一个活动相同,就称为他们处于同一个社交网络。求这n个人一共形成了多少个社交网络
基本思路
数据结构:
定义并查集数组father,写好相应的init、findFather、Union函数。
定义数组course,course[i]表示第i门课的课代表是coursr[i];
定义数组isRoot,下标为用户编号,值为这个社团的用户人数(只有下标为根用户的情况,值才不为0)。
对于每一个人,依次读入它所感兴趣的课程编号。如果该课程没有课代表,就将它自身作为课代表;然后将这个人和该课程课代表所在的集合合并。
统计根用户所在社团的用户人数。
统计跟用户的个数。
输出。
代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int MAXN = 1001;
int n;
vector<int> father;
vector<int> isRoot;//下标为某个社团代表人的编号,值为这个社团的人数
int course[MAXN] = {0};//下标为第i个爱好,值为这个爱好的代表人物编号
int cmp1(int a, int b){
return a > b;
}
//查找
int findFather(int v){
if(father[v]==v)
return v;
else{
int F = findFather(father[v]);
father[v] = F;
return F;
}
}
//合并
void Union(int a,int b){
int faA = findFather(a);
int faB = findFather(b);
if(faA!=faB){
father[faA] = faB;
}
}
int main(){
//读入人数,重新定义数组father和isRoot的大小
cin >> n;
father.resize(n + 1);
isRoot.resize(n + 1);
//初始化并查集
for (int i = 1; i <= n;i++){
father[i] = i;
}
//遍历1~n个人物
//读入人物i的爱好数量
//读入每个爱好的同时,检查这个爱好是否有代表人物
//如果没有,给这个爱好设置一个代表人物,就设置为当前人物i
//将当前人物i作为合并函数的一个固定参数,每个爱好的代表人物所在社团的代表人物作为合并另一个参数,调用合并函数
for (int i = 1; i <= n;i++){
int k;
scanf("%d:", &k);
for (int j = 1; j <= k;j++){
int ans;
cin >> ans;
if(course[ans]==0)
course[ans] = i;
Union(i, findFather(course[ans]));
}
}
//统计每个社团代表人所在社团的人数
//遍历1~n个人物,将isRoot[findFather(i)]++;
for (int i = 1; i <= n;i++){
isRoot[findFather(i)]++;
}
//统计社团个数
//遍历1~n个人物,如果isRoot[i]!=0,将cnt++
int cnt = 0;
for (int i = 1; i <= n;i++){
if(isRoot[i]!=0)
cnt++;
}
//将数组isRoot按人数逆序排序
sort(isRoot.begin(), isRoot.end(),cmp1);
//输出
cout<<cnt<<endl;
for(int i = 0; i < cnt; i++) {
printf("%d", isRoot[i]);
if(i != cnt - 1) printf(" ");
}
return 0;
}