并查集问题描述:
在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中,其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
定义
并查集是一种树型的数据结构,用于处理一些不相交集合(Disjoint Sets)的合并及查询问题。常常在使用中以森林来表示。
集就是让每个元素构成一个单元素的集合,也就是按一定顺序将属于同一组的元素所在的集合合并。
主要操作编辑
初始化
把每个点所在集合初始化为其自身。
通常来说,这个步骤在每次使用该数据结构时只需要执行一次,无论何种实现方式,时间复杂度均为O(N)。
查找
查找元素所在的集合,即根节点。
合并
将两个元素所在的集合合并为一个集合。
通常来说,合并之前,应先判断两个元素是否属于同一集合,这可用上面的“查找”操作实现。
【问题描述】若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。(人数≤5000,亲戚关系≤5000,询问亲戚关系次数≤5000)。
数据输入:
第一行:三个整数n,m,p,(n<=5000,m<=5000,p<=5000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。
以下m行:每行两个数Mi,Mj,1<=Mi,Mj<=N,表示Ai和Bi具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
数据输出:
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系
样例:
input
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
output
Yes
Yes
No
代码1:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;
int father[50002],a,b,m,n,p;
int find(int x){
if(father[x]!=x)
father[x]=find(father[x]);
return father[x];
}
int main()
{
int i;
scanf("%d%d%d",&n,&m,&p);
//初始化
for(i=1;i<=n;i++)
father[i]=i;
//合并两个元素所在的集合
for(i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
a=find(a);
b=find(b);
father[a]=b;
}
for(i=1;i<=p;i++)
{
scanf("%d%d",&a,&b);
a=find(a);
b=find(b);
if(a==b)//判断两个元素是否属于同一个元素
printf("Yes\n");
else
printf("No\n");
}
return 0;
}
代码2:
# include <iostream>
# include <cstdio>
using namespace std;
# define N 100
int father[N];
void init(int n);//初始化并查集
void union_(int x,int y);//合并两个元素所在的集合
bool same(int x,int y);//判断两个元素是否属于同一个集合
int getfather(int x);//获取根结点
int main(){
int n,m,p;//n个人,m个亲戚关系,p个测试用例,判断是否为亲戚
while(scanf("%d%d%d",&n,&m,&p)!=EOF){
int a,b;
init(n);
for(int i=0;i<m;i++){
scanf("%d%d",&a,&b);
union_(a,b);
}
for(int i=0;i<p;i++){
scanf("%d%d",&a,&b);
if(same(a,b)){
printf("YES\n");
}else{
printf("NO\n");
}
}
}
return 0;
}
void init(int n){
for(int i=1;i<=n;i++)
father[i] = i;
}
void union_(int x,int y){
x = getfather(x);
y = getfather(y);
if(x!=y)
father[x] = y;
}
int getfather(int x){
while(x!=father[x]){
x = father[x];
}
return x;
}
//每次查找,都把需要查找的点到根的路径上的所有点都记录为根
/*
int getfather(int x){
if(x!=father[x]){
father[x] = getfather(father[x]);// 路径压缩修改的是father数组
}
return father[x];
}
*/
bool same(int x,int y){
return getfather(x)==getfather(y);
}