临沂大学有很多社团,一个学生可能会加入多个社团。为了活跃大学生业余生活,增强体育运动积极性,临沂大学读书社团决定举行大学生跳绳比赛,要求该社团成员必须参加。为了扩大影响,要求只要其他社团有一个人参加,那么该社团中的每一个人都必须参加。求参加比赛至少多少人?
输入格式:
输入第一行包含两个整数n和m,n(0 < n <= 30000)表示学生的数目,m(0 <= m <= 500)表示社团的数目。每个学生都有一个唯一的编号,编号取值为0到n-1,编号为0的社团是读书社团。
接下来有m个社团的名单,每个社团的名单在输入中为一行。每一行先输入一个数k表示社团总人数。接着是社团中k个成员的编号。
输出格式:
输出一个数占一行,表示参加比赛的总人数。
输入样例:
100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
输出样例:
4
可以用并查集解决这道题,首先将每个社团的成员组成一个集合,因为一个学生可以加入不同的社团,所以接下来将读书社团的成员参与的其他社团进行合并,rank[find(i)]是i所在的集合人数,最后返回读书社团第一个人所在的集合的人数即可。
#include<bits/stdc++.h>
using namespace std;
vector<int>parent,ranks;
int find(int x){
return parent[x]==x?x:parent[x]=find(parent[x]);
}
void unite(int x,int y){
int rootx=find(x);
int rooty=find(y);
if(rootx!=rooty){
parent[rootx]=rooty;
ranks[rooty]+=ranks[rootx];
}
}
int main(){
int first=-1;//第一个社团第一个学生编号
int n,m;
cin>>n>>m;
parent.resize(n);
iota(parent.begin(),parent.end(),0);//parent初始化指向自己
ranks.resize(n);
fill(ranks.begin(),ranks.end(),1);//初始时每个集合只有一个人
for(int i=0,k;i<m;i++){
cin>>k;
for(int j=0,id,start=-1;j<k;j++){
cin>>id;
if(first==-1){
first=id;
}
if(start==-1){
start=id;
}
unite(start,id);//与社团的第一个人合并
}
}
cout<<ranks[find(first)];
return 0;
}
社团包含的学生,学生可以参加不同的社团,所有我们可以把社团和学生的关系看做是森林,节点是社团,社团包含的学生指向其他社团,使用广度优先搜索可以求出参与比赛的人数。flag代表是否被遍历过。
#include<bits/stdc++.h>
using namespace std;
struct Club;
struct Stu{
bool flag;
int id;
vector<Club*>club;
Stu(bool flag,int id):flag(flag),id(id){};
};
struct Club{
int num,id;
bool flag;
vector<Stu*>stu;
Club(bool flag,int id):flag(flag),id(id){};
};
int main(){
int n,m;
unordered_set<int>st;
cin>>n>>m;
vector<Stu*>stu(n);
for(int i=0;i<n;i++){
stu[i]=new Stu(false,i);
}
vector<Club*>club(m);
for(int i=0;i<m;i++){
club[i]=new Club(false,i);
}
for(int i=0;i<m;i++){
int num;
cin>>num;
club[i]->num=num;
for(int j=0;j<num;j++){
int id;
cin>>id;
stu[id]->club.push_back(club[i]);
club[i]->stu.push_back(stu[id]);
}
}
queue<Club*>que;
que.push(club[0]);
while(!que.empty()){
Club* t=que.front();
que.pop();
if(t->flag){
continue;
}
t->flag=true;
for(auto& _stu:t->stu){
if(_stu->flag){
continue;
}
_stu->flag=true;
st.insert(_stu->id);
for(auto& c:_stu->club){
if(c->flag){
continue;
}
que.push(c);
}
}
}
cout<<st.size();
return 0;
}