Week6 HomeWork 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

算法/思路分析

本题要求找到跟感染者有直接或间接接触过的所有人员，很明显可以使用并查集解决，因为并查集可以高效地查询两个人是否属于同一组（直接或间接接触过），而且可以合并两个人所在的组。

就本题而言，直接将每个组中的人两两合并，最后rnk[0]（0所在的组的大小）即为所有直接或间接接触过的人员。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath> 
using namespace std;
const int maxn = 3e4+10;
#define ll long long
int par[maxn], rnk[maxn];
int pp[maxn];
int n,m;
void init()
{
	for(int i = 0; i < n; i++)
	{
		par[i] = i;
		rnk[i] = 1;
	}
	memset(root,0,sizeof(root)); 
}
//并查集
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(x < y) swap(x,y);
	par[x] = y; 
	rnk[y] += rnk[x];
	rnk[x] = rnk[y];
	return true;
}

int main()
{
	while(1)
	{
		scanf("%d%d",&n,&m);
		if(n == 0 && m == 0) break;
		init();
		for(int i = 1; i <= m; i++)
		{
			int num;
			scanf("%d",&num);
			for(int i = 0; i < num; i++) scanf("%d",&pp[i]);
			for(int i = 1; i < num; i++)
			 	unite(pp[i-1],pp[i]);
		}
		printf("%d\n",rnk[0]);
	}

	return 0;
}