并查集详解
一、并查集
1.定义
首先,并查集是个啥?
并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中,其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。并查集是一种树型的数据结构,用于处理一些不相交集合(Disjoint Sets)的合并及查询问题。常常在使用中以森林来表示。--度娘
我们可以用一个更通俗一些的例子来理解:
有a,b,c三个人,
b是a的上司,a又是c的上司,但是c却不认识b,这显然是不合理的,所以我们要让b变成最大的上司。
首先我们定义一个f数组来维护这段关系,f[a]=b表示a的上司是b,f[c]=a表示c的上司是a,那如何将c的最大上司变成b呢?
定义一个find函数
//初始化每个人的上司是自己,即令f[i]=i int find(x){ if(f[x]==x) return x; return find(f[x]);//递归找到自己最大的上司 } f[c]=find(f[a]);//令b变为c的上司
这样我们就解决了这个问题
ps:为了符合规定,一个人不能有多个上司(好奇怪哦
2.路径压缩
如果b有上司,b的上司也有上司,上司的上司还有上司...
可以发现如果这段关系有很多层,那么我们需要递归很多次才能找到c的最大上司x,时间复杂度可能会很高,所以我们可以用路径压缩优化
我们将c到x中间的那些节点删去,直接令f[c]等于c的最大上司x,这样便能够大大提高效率
代码如下:
int find(x){ if(f[x]==x) return x; return f[x]=find(f[x]); }
二、例题
1.洛谷P3367 【模板】并查集
https://www.luogu.org/problem/P3367
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数N、M,表示共有N个元素和M个操作。
接下来M行,每行包含三个整数Zi、Xi、Yi
当Zi=1时,将Xi与Yi所在的集合合并
当Zi=2时,输出Xi与Yi是否在同一集合内,是的话输出Y;否则话输出N
输出格式
如上,对于每一个Zi=2的操作,都有一行输出,每行包含一个大写字母,为Y或者N
输入输出样例
输入 #14 7 2 1 2 1 1 2 2 1 2 1 3 4 2 1 4 1 2 3 2 1 4输出 #1N Y N Y说明/提示
时空限制:1000ms,128M
数据规模:
对于30%的数据,N<=10,M<=20;
对于70%的数据,N<=100,M<=1000;
对于100%的数据,N<=10000,M<=200000。
模板题,直接上代码
#include<bits/stdc++.h> using namespace std; int f[100001],n,m,a,b,c; int find(int x){//并查集函数 if(f[x]==x) return x; return f[x]=find(f[x]); } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) f[i]=i;//初始化,自己是自己的上司 for(int i=1;i<=m;i++){ scanf("%d%d%d",&a,&b,&c); if(a==1) f[find(b)]=find(c);//合并,令c的上司成为b的上司的上司 else{ if(find(b)==find(c)) cout<<"Y"<<endl;//判断两人的上司是否为同一个人并输出 else cout<<"N"<<endl; } } return 0; }
持续更新ing...