B-戴好口罩(并查集的应用)

题意:

如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。

输入要求:
多组数据。每组数据第一行为两个整数 n 和 m(n = m = 0表示输入结束,不需要处理),n是学生数量,m是学生群体数量,0 < n <= 3e4 , 0 <= m <= 5e2。
学生编号为 0 ~ n - 1;小 A 同学的编号为 0。
之后输入m行,每行有一个整数 num 表示小群体人数,随后又 num 个整数代表该群体的学生编号。
输出要求:
输出要隔离的人数,每组数据的答案输出占一行
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

思路:

  • 该题是个显然的并查集应用,所有接触过的人都放在一类人中。每个接触都对并查集进行 num 次 unite 操作。
  • 并查集使用一个数组 par[n] 来表示,其中查找函数 find(int x) 是一个递归函数,若 par[x] == x,则返回 par[x],反之则返回 find(par[x]),并且将 find(par[x]) 赋值给 par[x]。合并函数 unite(int x, int y) 函数,首先找到两者各自的代表数 x_ 和 y_,在将两者合并(par[y_] = x_;),由于该题需要输出隔离的人数,即输出某个集合的元素个数,故又定义了一个数组 rnk 用于记录集合的人数。

代码:

#include <iostream>
using namespace std;

int par[30005], rnk[30005];
int group[30005];//用于存放一个小团体的人

int find(int x) {
	if (par[x] == x)return par[x];
	return par[x] = find(par[x]);
}
void unite(int x, int y) {
	int x_ = find(x);
	int y_ = find(y);
	if (x_ == y_)return;
	if (x_ < y_) {
		par[y_] = x_;
		rnk[x_] = rnk[x_] + rnk[y_];
	}
	else {
		par[x_] = y_;
		rnk[y_] = rnk[y_] + rnk[x_];
	}
}

int main()
{
	int n, m, num;
	while (1) {
		scanf_s("%d%d", &n, &m);
		if (m == n && m == 0)break;
		for (int i = 0; i < n; i++) {
			par[i] = i;
			rnk[i] = 1;
		}
		for (int i = 0; i < m; i++) {
			cin >> num;
			for (int j = 0; j < num; j++) {
				cin >> group[j];
				if (j > 0) {
					unite(group[j - 1], group[j]);
				}
			}
		}
		cout << rnk[0] << endl;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值