您的位置:首页 > 其它

POJ 1611 / ZOJ 1789: The Suspects - 并查集

2015-01-10 21:41 435 查看
学姐给我这个菜鸟讲的题让我第一次接触到并查集的使用。
本题是并查集的基础应用题。
题目大意:SARS(非典型肺炎)传播得非常厉害,其中最有效的办法是隔离那些患病、和患病者接触的人。现在有几个学习小组,每小组有几个学生,一个学生可能会参加多个小组。小组中只要有一个人得病,其余的都是嫌疑人。现在已知这些小组的人员,且0号学生已经患病,求一共有多少个嫌疑人。分析:每个小组的人员属于同一个集合。在根节点记录每个集合的个数num[i],然后找到0所属的根节点,num[findSet(0)]即为所有的嫌疑人数。
输入:每个案例第一行为志愿者人数n和组成小组数量m,接下来是n行,每行第一个数k是当前小组的成员人数,其余数为各个小组中成员的编号。
感觉这个同志的blog里写的很详细:http://cavenkaka.iteye.com/blog/1171373
#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int Max=30001;

int p[Max],rank[Max];
int num[Max];

int Find_Set(int x) //路径压缩:经过"递推"找到祖先节点后,"回溯"的时候顺便将它的子孙节点都直接指向祖先
{
if(x!=p[x])
p[x]=Find_Set(p[x]);
return p[x];
}
void Union(int x,int y) //按秩合并:合并的时候将元素少的集合合并到元素多的集合中,这样合并之后树的高度会相对较小
{
x=Find_Set(x);
y=Find_Set(y);
if(x==y)
return;
if(rank[x]>rank[y])
{
p[y]=x;
num[x]+=num[y];
}
else
{
p[x]=y;
if(rank[x]=rank[y])
rank[x]++;
num[y]+=num[x];
}
}
int main()
{
int i,j,k,n,m;
int a,b;
int s1,s2;
while(true)
{
scanf("%d %d",&n,&m);
if(n==0&&m==0)
break;
for(i=0;i<n;i++)
{
p[i]=i;
rank[i]=0;
num[i]=1;
}
for(i=0;i<m;i++)
{
scanf("%d%d",&k,&a);
for(j=1;j<k;j++)
{
scanf("%d",&b);
s1=Find_Set(a);
s2=Find_Set(b);
Union(s1,s2);
}
}
i=Find_Set(0);
printf("%d\n",num[i]);
}
return 0;
}

********************************************************************************************************************************************************************************************************************************************1. 合并两个集合void merge(int a, int b) {par[get_par(a)] = get_par(b);}
2. 查询一个元素在哪个集合int get_par(int a) {if ( par[a] != a ) par[a] = get_par(par[a]);return par[a];}
3. 查询两个元素是否属于同一集合int query(int a, int b) {return get_par(a) == get_par(b);}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: