week6作业题_B - 戴好口罩！

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

题目思路

根据题意可知，这是一个简单的并查集的问题。其中并查集集合为par，初始化的时候令par[i] = i即n个孤立元素，独自一组。在这里面并查集的实现时，我使用了路径压缩的方法，即将每个一组中的节点都与它的代表元素直接相连。在相连的时候如果两个元素不是属于同一组的话，就将两个集合连接，这里我们为了避免变成"高树"，我们使用一个rnk数组来维护集合的大小。

代码实现

//#include <bits/stdc++.h>
#include <iostream>
#include <algorithm>
#define _for(i,a,b) for(int i = (a); i < (b); i++)
#define _rep(i,a,b) for(int i = (a); i <= (b); i++)
using namespace std;
const int MAXN = 3e4 + 10;

int n,m,a,b,num;
int par[MAXN],rnk[MAXN];	//并查集集合，记录集合的大小 

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()
{
	ios::sync_with_stdio(false);		//关闭同步
	while(cin >> n >> m)
	{
		if(n == 0 && m == 0) break;
		_for(i,0,n)
		{
			par[i] = i;
			rnk[i] = 1;
		}
		_for(i,0,m)
		{
			cin >> num;
			cin >> a;
			_for(j,1,num)
			{
				cin >> b;
				unite(a,b);
			}
		}
		cout << rnk[find(0)] << endl;
	}
}