B-戴好口罩！_b - 戴好口罩!-CSDN博客

本文链接：https://blog.csdn.net/weixin_43526172/article/details/105176465

B-戴好口罩！

一、题目描述

新型冠状病毒肺炎（Corona Virus Disease 2019，COVID-19），简称“新冠肺炎”，指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体，那么这个群体需要被隔离！
小A同学被确诊为新冠感染，并且没有戴口罩！！！！！！
危！！！
时间紧迫！！！！
需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。
众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。
请你编写程序解决！戴口罩！！

Input

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

Output

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

Sample Input

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output

4
1
1

二、思路与算法

本题核心算法为并查集。
通过并查集实现初始化init、查询find、合并unite三个功能，即足以解答本题。
每个群体中的各个学生，都合并在一个同并查集中，最后查询0号学生所在并查集中共有多少人即可。

并查集初始化init：uni[i]=i rank[i]=1，初始每个元素自成一个并查集，每个并查集的代表元是自己，并查集大小都是1。
并查集查询find：递归，直到找到所在并查集的代表元。
并查集合并unite：找到两个元素所在并查集的代表元，如果不同，把小并查集纳入大并查集中（改动小并查集代表元的uni值，改大并查集大小）。

三、代码实现

#include<cstdio>
#include<iostream>
using namespace std;

int n=0,m=0,tot=0;
int uni[200000]={-1};
int rank[200000]={-1};
int x=-1,cur=-1;

void init(){
	for(int i=0;i<n;i++){
		uni[i]=i;	rank[i]=1;
	}
}

int find(int x){
	if(uni[x]==x){
		return x;
	}
	return uni[x]=find(uni[x]);
}

bool unite(int x,int y){
	x=find(x);	y=find(y);
	if(x==y){return false;	}
	if(rank[x]>rank[y]){	swap(x,y);} 
	uni[x]=y;   //把x挂到y上 
	rank[y]=(rank[x]+=rank[y]);
	//printf("after unite %d rank is %d\n",x,rank[x]);
	
	return true;
}

int main(){
	scanf("%d %d",&n,&m);
	while(n!=0||m!=0){
		init();				//初始化 
		for(int i=0;i<m;i++){  //m个组 
			scanf("%d",&tot);  //组内元素数目 
			scanf("%d",&x);
			for(int j=1;j<tot;j++){
				scanf("%d",&cur);
				//printf("unite %d %d\n",x,cur);
				unite(x,cur);
				x=cur;
			}
		}
		printf("%d\n",rank[find(0)]);
		scanf("%d %d",&n,&m);
	}
	return 0;
}