Week6 作业 B - 戴好口罩！ POJ - 1611 并查集

本文链接：https://blog.csdn.net/ziseon/article/details/105133937

题目描述

新型冠状病毒肺炎（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

解题思路

本题用并查集解，将同一小组的同学在并查集中合并，最后输出和0号同学同一并查集同学的数量。

程序源码

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<stdio.h>
using namespace std;

int stu[35000]; //并查集
int sco[35000]; //总人数

int search1(int num) { //并查集搜索
	if (stu[num] == num) { //找到了
		return num;
	}
	stu[num] = search1(stu[num]); //没找到，递归搜索
	int root = stu[num];
	
	int a;
	while (num != root) { //路径压缩
		a = stu[num];
		stu[num] = root;
		num = a;
	}
	return root; //返回
}
void unite(int a, int b) { //并查集合并
	int fa = search1(a); //找到a的根
	int fb = search1(b); //找到b的根
	if (fa != fb) { //将fa指向fb
		stu[fa] = fb;
		sco[fb] += sco[fa]; //累加计数
		return;
	}
	return;
}
int main() {
	int n, m, c, last = -1, cn = 0;
	while (scanf("%d%d", &n, &m)) { //获取输入
		if (n == 0 && m == 0) {
			return 0;
		}
		for (int i = 0; i <= n; i++) { //初始化并查集
			stu[i] = i;
			sco[i] = 1;
		}
		for (int i = 0; i < m; i++) {
			scanf("%d", &c); //每组的人数
			last = -1;
			for (int j = 0; j < c; j++) { //并查集合并
				scanf("%d", &cn);
				if (last == -1) last = cn;
				else {
					unite(last, cn);
					last = cn;
				}
			}
		}
		int rs = search1(0);
		cout << sco[rs] << endl; //输出和0号同学同一并查集的人数
	}
	return 0;
}