关于并查集
并查集(disjoint-set),在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并
我是一个彩蛋 :DDD
一个数据结构
如下
如图是一个...
不是,放错了 QAQ
是这个
如图是一个简单的树 -w- 这个树有三层,简单理一下:
在这一个树中,2和3是1的孩子,4 5是2的孩子,6 7是3的孩子
所以2 3 4 5 6 7 都是1的孩子
而1是2 3 4 5 6 7的祖父 :P
查找
而并查集的查找主要是找到一个树里面的数的根节点
也就是往上找,一直找到根
比如要找4所在的树的根,就先找到2,再向上找到1,在想往上找的时候发现没有的找了,于是4所在的树的根是1
这个...很简单 。看着就知道是一个递归或者循环搞定的吧 :D
等等 好像还没讲存储吧 怎么就讲上查找了 Σ(っ °Д °;)っ
咳咳...
存储
先看看这张图片
节点有:1,2,3,4,5,6,7
而并查集是用一维数组实现的...
怎么样,想到了数组的下标?
(qwqwqwqwqwq
对了,所以并查集会用数组下标来表示并查集中的数 (喜欢用0开始当下标的注意了 :O
(别担心,题目都是给1234这种的1开始有顺序的数字 :O 待会让你见识见识 :P
那数组存数怎么搞🤔?
别急
首先已经知道用上数组了,那数组最基础的功能肯定是存数啊 qwq
关于并查集,重要的有两个 :一个是数,一个是这个数的父节点
所以我们用数组来存这个数的父节点 (太直白了吧!!
看看过程:
再看看上面的树🤔
由图得
2的父节点是1;3的父节点是1;4和5的父节点是2;6和7的父节点是3
得到如图
那么1那里存什么呢🤔 总不能什么都不存吧
1的父节点就是他自己辣
所以当我们看到一个下标中存的数和这个下标一样时,这个数就是父节点了
在找的时候,比如要找7所在的树的树跟
- 找数组中下标为7储存的数 是3
- 那就再找下标为3储存数 是1
- 再找下标为1的数 是1 就是这个数的父节点是自己 这个数就是7所在的树的树跟了
- 就找到了🎉🎉🎉🎉🎉🎉🎉🎉🎉
并查集的合并
虽然开头就讲了并查集的概念
是把不同集合的元素合并
如图
有一个8带着自己的俩小弟(9和10)过来,说要认1当大哥 (orz
所以1和8就会扯上关系 我们就把他们撮合在一起
如图
在1和8扯上关系前 表示并查集的数组长这样
如图,因为8还没跟1扯上关系 所以8是自己的父节点 就是自己那里的小团体的牢大
所以我们找9和10所在的树的根的时候 是直接找到8头上的 不会找到1头上
然后,当1和8扯上关系时 数组长这样
唯一的变动就是下标8中存的数便成了1
因为8认1当了老大,所以8的父节点就是1
这样找8,9,10的根的时候就会找到1了
这样两个并查集就合并辣🎉🎉🎉🎉🎉🎉
并查集一般都是把两个集合的根合并起来
关于并查集的基本代码
查找根的代码
关于查找根,就是刚才最先说的
也就是往上找,一直找到根
这里我们用递归
int find(int x){
//x是要找的下标
//f是并查集数组
if(f[x]==x) return x;//找到自己是自己父节点的数,就是根,于是返回这个数
else return find(f[x])//不是就把当前的数(f[x])当下标继续找父节点 直到找到根
}
很简单吧 :D
下边是究极无敌进阶版
int find(int x)
{
return f[x]==x?x:find(f[x]);
}
有一点点的抽象,对吧?
看那一行代码中 都看的懂 就是连在一起看不懂了 :(
以下是原理:
总的来说,returen后面的那一串东西就是在选return什么
仔细看,比较难懂就是那一个?和:
这其实和if-else是一个东西,?是充当if,而:是充当else
这一串代码是分成了两个部分
f[x]==x?x 和:find(f[x])
第一个部分,f[x]==x?x
写通俗点就是 if(f[x]==x) return x; (也就是?前面的是判断条件,而?后面的是执行的语句
第二部分:find(f[x])就是 else return find(f[x]); (也就是:是起到else的作用的,而:后面的是执行的
整串代码来说的话 就是如果f[x]==x是true的话就让return返回x的值否则就返回find(f[x])的值
感觉讲不怎么明白 这是条件运算符可以去搜一下 :L
合并的代码
合并有一个前提:就是两个集合不能是同一个根 (废话
好吧,虽然是废话,但我们还是要判断一下的
如下是代码
void L(int x,int y)
{
int fx=find(x);//找要合并的两个数的根
int fy=find(y);//也就是fx是x所在的树的根,而fy是y的根
if(fx!=fy) f[fx]=fy;//如果两个数不是同一个根,就是不是同一个集合 就把他们合并
//f[fx]=fy 是把x所在的集合的根融合到y所在的集合的根上
}
(其实合并就是相当于存储了叭 :P
初始化
在程序开始时,我们总是把每个元素当成自己的父节点,也就是程序开始时是如下
程序开始时他们互不关联,在程序后面才渐渐把他们合起来
把每个元素当成自己的父节点
像这样:
这样做,非常的简单 :D
不就一个f[i]=i搞定的嘛🤔
所以初始化并查集的代码如下
for(int i=1;i<=a;i++)//a是接下来输入的并查集的点的数目,题目会给(待会康康题 :P
{
f[i]=i;//把每一个数组元素等于其下标 也就是每一个数都是根
}
求有几个集合
并查集一般会长成像这样:
有好几个不同的集合
这时,要求出有几个集合,虽然用图一目了然,但是程序只会看数组不会看图啊 :L
再仔细看看,一个集合只有一个根,那不是只用找有几个根就🆗了!!
而根有几个,不是只要看有几个自己是自己父节点的数不就🆗了!!
根据以上得出代码如下
int root()
{
int c=0;//初始化计数器
for(int i=1;i<=a;i++)
{
if(f[i]==i) c++;//一个数为自己的父节点的时候,这个数就是根,于是计数器+1
}
return c;//返回根的数量
}
(其实一道题也用不了几次,不用函数直接搬到主代码里也行 :O
以下是一些题目
一些小题
A. 是不是亲戚
题目链接:1921 - 是不是亲戚-东方博宜OJ
题目描述
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚y的亲戚也都是x的亲戚。
输入
第一行:三个整数 n,m,p(n≤5000,m≤5000,p≤5000 ),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。
以下m行:每行两个数 Mi,Mj;1≤Mi,Mj≤N,表示 Mi 和 Mj 具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问 Pi 和 Pj 是否具有亲戚关系。
输出
P行,每行一个 Yes
或 No
。表示第 i 个询问的答案为“具有”或“不具有”亲戚关系。
样例
输入
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
输出
Yes
Yes
No
代码
#include<bits/stdc++.h>
using namespace std;
int f[5010];
//并查集数组要定义全局,因为函数要用到的最多
int find(int x)
{
//查找根的函数
if(f[x]==x) return x;
else return find(f[x]);
}
void L(int x,int y)
{
//合并函数
int fx=find(x);
int fy=find(y);
if(fx!=fy) f[fx]=fy;
}
int main(){
int n,m,p;
cin>>n>>m>>p;//定义和输入n,m,p
for(int i=1;i<=n;i++)
{
f[i]=i;
}
//初始化并查集数组
int x,y;
for(int i=1;i<=m;i++)
{
cin>>x>>y;//输入m行亲戚关系
L(x,y);//使用超级函数合并,把x的根连在y的根上
}
for(int i=1;i<=p;i++)
{
cin>>x>>y;//输入接下来的p行询问
//x,y前面输入的已经在并查集数组里了,所以可以拿来用 :P 新定义一个也不是不行
if(find(x)==find(y)) cout<<"Yes"<<endl;//如果是同一个根,也就是是亲戚,就输出Yes
else cout<<"No"<<endl;//不是就输出No
}
return 0;
}
解析
这道题是一道最²²²基础的并查集题目,先来理解1下题目:
先像我前面说的题目会给有几个点,就是题目中的n,然后是下面输入的m行和接下来询问的p行
重要的是中间的m行的关系,要通过给的这些关系把并查集中n个的点联系起来,拿样例来说:
程序开始是是这样
输入1和2,再看看合并的函数,是把x的根融到y的根上,所以输入1和2后的数组长这样输入1和5,先找1的根是2,然后5的根是5,就把2融到5上,如图:
然后后面的同理... (注意是不同的根才融合
全部完5行关系后,数组长这样:
然后在输入p行询问:询问输入的两个点是否有关系
这个...呃,是同一个根的就必定在同一个集合里,也就是有关系
所以我们只要找询问的两个数的根是否相同就🆗了🎉🎉🎉🎉🎉🎉🎉
这是上面输入的样例的询问,可以自己看上面的图片对一下 :P
1 4
2 3
5 6
B. 修路
题目描述
某市调查城镇交通状况,得到现有城镇道路统计表。表中列出了每条道路直接连通的城镇。市政府 "村村通工程" 的目标是使全市任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要相互之间可达即可)。
请你计算出最少还需要建设多少条道路?
输入
输入包含若干组测试数据,每组测试数据的第一行给出两个用空格隔开的正整数,分别是城镇数目n和道路数目m(输入n=0表示测试数据结束);
随后的m行对应m条道路,每行给出一对用空格隔开的正整数,分别是该条道路直接相连的两个城镇的编号。简单起见,城镇从1到n编号。
注意:两个城市间可以有多条道路相通。
对于100%的数据,保证10001≤n<1000 。
输出
对于每组数据,对应一行一个整数。表示最少还需要建设的道路数目。
样例
输入
4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0
输出
1
0
2
998
看题
简单看看题,会给城镇的数量,也就是点的数量
然后会给到道路,就是哪两个城镇有道路联通
代码
#include<bits/stdc++.h>
using namespace std;
int f[1010];
//定义并查集数组,因为函数要用 :P
int n,m;
//在全局定义n,m变量是因为下面的root函数要用
int find(int x)
{
//找根的函数
if(x==f[x]) return x;
else return find(f[x]);
}
void L(int x,int y)
{
//合并的函数
int fx=find(x);
int fy=find(y);
if(fx!=fy) f[fx]=fy;
}
int root()
{
//找有多少个根 QwQ
int k=0;
for(int i=1;i<=n;i++)
{
if(f[i]==i) k++;
}
return k;
}
int main(){
while(true){
//while(true):不到退出不结束的循环 :D
cin>>n;//输入n个城镇,就是接下来的并查集的点的数量
if(n==0) break;//如题,输入n=0表示程序结束
cin>>m;//输入m条道路(接下来有关系的城镇),也就是接下来输入m行
for(int i=1;i<=n;i++)
{
f[i]=i;
}
//浅浅初始化一下并查集数组
int x,y;
for(int i=1;i<=m;i++)
{
cin>>x>>y;
L(x,y);
//简单的合并,应为待会要计算根的个数
}
cout<<root()-1<<endl;
//根的个数减一就是要连接的路个数
//这个...呃..很好理解吧 三个点只用两个线连接 :P
}
return 0;
}