并查集-The Suspects POJ - 1611
题目:
2003年4月16日,世界卫生组织根据包括中国内地和香港地区,加拿大、美国在内的11个国家和地区的13个实验室通力合作研究的结果,宣布重症急性呼吸综合征的病因是一种新型的冠状病毒,称为SARS冠状病毒。
很不幸,小明同学昨天晚上被确诊以经感染非典病毒,为此,校医院需要对他隔离治疗并隔离观察与他直接或间接接触者。
可以认为,同一个社团内如果有人感染病毒或可能感染病毒,那么这个社团内所有人都被认为是可能已经感染了病毒,由于时间紧迫,需要尽快找到所有可能携带病毒的同学并隔离,以防止更大范围的病毒扩散,院长请你帮忙编写程序计算需要隔离多少人
Input
输入文件包含多组数据。
对于每组测试数据:
第一行为两个整数n和m
其中n是学生的数量,m是学生社团的数量。0 < n <= 30000 , 0 <= m <= 500
每个学生编号是一个0到n-1之间的整数,已知小明同学编号是0
紧随其后的是团体的成员列表,每组一行。
每一行有一个整数k,代表成员数量。之后有k个整数代表这个群体的学生。一行中的所有整数由至少一个空格隔开。
n = m = 0表示输入结束,不需要处理。
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
题意:
输入点的数量n,点集的数量m
输入m个点集中点的数量k,以及k个点的编号
输出与0号点所在点集的剩余点的数量
思路:
采用并查集。
每读入一个新点都将前一个点的祖先跟在新点的祖先下面。
最后求和0号点相同祖先的点的数量。
注意路径压缩。
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=3e4+5;
int n,m,pre[maxn];
int Find(int root)
{
if(root==pre[root]) return root;
return pre[root]=Find(pre[root]);//路径压缩的递归写法
}
int Find1(int root)//路径压缩的迭代写法
{
int son,tmp;
son=root;
while(root!=pre[root])
root=pre[root];//root先指向祖先
while(son!=root)//把root原先所在的位置以及其上级直接指向祖先
{
tmp=pre[son];
pre[son]=root;
son=tmp;
}
return root;
}
int Find2(int root)
{
while(root!=pre[root])
root=pre[root];
return root;
}
int Find3(int root)
{
if(root==pre[root]) return root;;
root=Find3(pre[root]);
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
if(n==0&&m==0) return 0;
//memset(pre,-1,sizeof(pre));
for(int i=0;i<n;i++) pre[i]=i;//不能用memset成同一值,否则死循环
for(int i=0;i<m;i++)
{
int num,root;//root即每个社团的主席(祖先)
scanf("%d%d",&num,&root);//先输入个root
for(int j=1;j<num;j++)
{
int cnt;
scanf("%d",&cnt);
pre[Find1(root)]=Find1(cnt);//是Find(root)而不是root,因为多个社团可能共有同一个人,root的祖先跟在cnt的祖先之下
root=cnt;//更新一下root
}
}
int ans=0;
for(int i=0;i<n;i++)
if(Find1(pre[i])==Find1(0))
ans++;
printf("%d\n",ans);
}
return 0;
}
总结:
- pre数组初始化为各点本身,不能为同一值
- return pre[root]=Find(pre[root]);//路径压缩的递归写法