c++并查集模板题——戴好口罩

题意:

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

思路

分组,将所有接触过的人分成一组,并查集模板题。
顺便维护一个人数,以便最后输出有多少人。

代码

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
using namespace std;
//并查集
int par[30005],sum[30005];   //par 并查集  sum 人数
void ini(int n)
{
	for(int i=0;i<=n;i++)
	{
		par[i]=i;    //每个人单独分组
		sum[i]=1;   //每组一个人
	}
} 
int find(int x)
{  //查
	if(par[x]==x) 
		return x;      
	return par[x]=find(par[x]);    //优化 所有元素都指向代表元素
}
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)    //如果不在一组 则合并
	{
		par[x]=y;
		sum[y]+=sum[x];   //合并的时候人数相加
	}
}
int main()
{
	int n,m;
	while(scanf("%d %d",&n,&m))
	{
		ini(n);
		if(n==0&&m==0) return 0;
		while(m--)
		{
			int num,x,p=-1;
			scanf("%d",&num);
			while(num--)
			{
				scanf("%d",&x);
				if(p==-1) p=x;
				else
				{
					unite(x,p);
					p=x;
				}
			} 
		 } 
		printf("%d\n",sum[find(0)]);
	} 
	return 0;
}
发布了25 篇原创文章 · 获赞 1 · 访问量 694
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 游动-白 设计师: 上身试试

分享到微信朋友圈

×

扫一扫,手机浏览