SDU程序设计思维与实践Week6 B 戴好口罩！【并查集】

B 戴好口罩！

输入

多组数据，对于每组测试数据：
第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

输出

输出要隔离的人数，每组数据的答案输出占一行

思路分析

并查集即可。注意合并时小树合并到大树。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=3e4+5; 
int par[maxn],rnk[maxn];
//并查集
void init(int n){
	for(int i=0;i<n;i++){
		par[i]=i,rnk[i]=1;		
	}
} 
int find(int x){
	return par[x]==x? x: par[x]=find(par[x]);
}
bool unite(int x,int y){
	x=find(x),y=find(y);
	if(x==y) return false;
	if(rnk[x]>rnk[y]) swap(x,y);
	par[x]=y;rnk[y]=(rnk[x]+=rnk[y]);
	return true;
}
int main(){
	int n,m;
	while(~  scanf("%d%d",&n,&m)&&!(n==0&&m==0) ){
		init(n);
		while(m--){
			int num,last=-1;
			scanf("%d",&num);
			while(num--){
				int temp;
				cin>>temp;
				if(last!=-1){
					unite(last,temp);
				}
				last=temp;
			}
		}
		cout<<rnk[find(0)]<<endl;
	} 
}