作用与基本原理:
套路问题:
用一道模板题方便讲解;
合并集合
首先设置一个函数find();
find()函数的作用如问题2, 求集合的祖宗节点
(while(p[x]!=x) x=p[x] 如果不是根就找上一个,直到找到根为止)
元素合并操作:(元素也是集合)
开始时每个集合都是一个独立的集合,并且都是等于自己本身下标的数
例如:
p[5]=5,p[3]=3;
如果是M操作的话那么就将集合进行合并,合并的操作是:
p[3]=p[5]=5;
所以3的祖宗节点便成为了5,此时以5为祖宗节点的集合为{5,3}
如果要将p[9]=9插入到p[3]当中,应该找到3的祖宗节点,然后再把p[9]=9插入其中,所以p[9]=find(3);(find()函数用于查找祖宗节点)
集合合并操作:
假设有以6为祖宗节点的集合为{6,4,7,10}。如果要将以6为祖宗节点的集合插入到以5为祖宗节点的集合,则该操作可以是p[6]=find(3)(或者find(9),find(5));
此时p[6]=5
当然如果是以6为祖宗节点集合中的4,7,10则可以这样
p[find(4)]=find(3)
总结:
合并操作:p[以a为祖宗节点的集合]=以b为祖宗节点的集合; (以a为祖宗节点的集合插到以b为祖宗节点的集合)
但是如果每次查找祖宗节点都要while一遍(while(p[x]!=x) x=p[x]),耗时大,于是就有了路径压缩;
int find(int x){ //祖先节点的父节点是自己本身 if(p[x] != x){ //将x置为x的祖先节点 p[x] = find(p[x]); } return p[x]; }
路径压缩具体操作:
如图1
find(1) p[1] = 2 p[1] = find(2)
find(2) p[2] = 3 p[2] = find(3)
find(3) p[3] = 4 p[3] = find(4)
find(4) p[4] = 4 将p[4]返回
于是一路回溯;4=p[4]=p[3]=p[2]=p[1];
注意只需要查找一遍,以后要用时间就是O(1)了
#include<iostream> using namespace std; int p[100010]; int find(int x) { if(p[x]!=x) p[x]=find(p[x]); //核心操作 return p[x]; } int main() { int n,m; cin>>n>>m; for(int i=1;i<=n;i++) p[i]=i; while(m--) { char op; int a,b; cin>>op>>a>>b; if(op=='M') p[find(a)]=find(b); //集合合并,将集合a插到集合b else { if(find(a)==find(b)) printf("Yes\n"); else printf("No\n"); } } return 0; }
总结:利用根相同特性查找集合
连通块中点的数量
#include <iostream> using namespace std; const int N = 100010; int n,m; int p[N],cnt[N]; int find(int x) { if(p[x]!=x) p[x]=find(p[x]); return p[x]; } int main() { cin>>n>>m; for(int i=1;i<=n;i++) { p[i]=i; cnt[i]=1; //将每个元素初始化为1 } while(m--) { string op; int a,b; cin>>op; if(op=="C") { cin>>a>>b; a=find(a),b=find(b); p[a]=b; if(a!=b) //对不同集合合并才相加 { cnt[b]+=cnt[a]; } } else if(op=="Q1") { cin>>a>>b; if(find(a)==find(b)) puts("Yes"); else puts("No"); } else { cin>>a; cout<<cnt[find(a)]<<endl; } } return 0; }
总结:利用两个集合根不等特性相加求和
蓝桥杯2017年第八届真题-合根植物
w星球的一个种植园,被分成 m * n 个小格子(东西方向m行,南北方向n列)。每个格子里种了一株合根植物。
这种植物有个特点,它的根可能会沿着南北或东西方向伸展,从而与另一个格子的植物合成为一体。
如果我们告诉你哪些小格子间出现了连根现象,你能说出这个园中一共有多少株合根植物吗?
输入格式
第一行,两个整数m,n,用空格分开,表示格子的行数、列数(1<m,n<1000)。
接下来一行,一个整数k,表示下面还有k行数据(0<k<100000)
接下来k行,第行两个整数a,b,表示编号为a的小格子和编号为b的小格子合根了。
格子的编号一行一行,从上到下,从左到右编号。
比如:5 * 4 的小格子,编号:
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20
样例输入
5 4 16 2 3 1 5 5 9 4 8 7 8 9 10 10 11 11 12 10 14 12 16 14 18 17 18 15 19 19 20 9 13 13 17
样例输出
5
其合根情况参考下图:
如图所示,连一条线格子就少1,合并一次就将m*n减1一次,剩下的就是独立的
#include <iostream> using namespace std; int n,m,k; int p[1000010];//要1e6 1000*1000 int find(int x) { return x==p[x]?x:p[x]=find(p[x]); } int main() { cin>>n>>m>>k; int root=n*m; for(int i=1;i<=root;i++) p[i]=i; while(k--) { int a,b; cin>>a>>b; if(find(a)!=find(b)) { p[find(a)]=find(b); root--; } } cout<<root; return 0; }
总结:利用最多操作初始化次的特性(操作root次后就只剩一个集合,后续操作无效),求集合个数
蓝桥杯][2019年第十届真题] 修改数组
思路:
每查到一个数,就将这个数的并查集树插到比他高一个节点的树上;这样就可以保证每次查到树上元素时输出树顶元素;
如 :查到2,则将2插到比他高一个节点的树上即3;查到1,则将1插2上;由并查集原理,1,2,3的值都指向树顶元素3;那么再查1时,就会输出3,并且将1,2,3这颗树插到4上,构成以元素4为树顶的树
#include <iostream> using namespace std; const int N=1e6+10; int p[N]; int find(int x) { return x==p[x]?x:p[x]=find(p[x]); } int main() { int n; cin>>n; for(int i=1;i<=N;i++) p[i]=i; for(int i=0;i<n;i++) { int m; cin>>m; m=find(m); cout<<m<<" "; p[m]=m+1; } return 0; }
蓝桥幼儿园
输出描述
对于每个 op=2 的输入,如果 x 和 y 是朋友,则输出一行 YES
,否则输出一行 NO
。
输入输出样例
输入
5 5 2 1 2 1 1 3 2 1 3 1 2 3 2 1 2
输出
NO YES YES
#include <iostream> using namespace std; int p[200010]; int find(int x) { return x==p[x]?x:p[x]=find(p[x]); } int main() { int n,m,op; cin>>n>>m; for(int i=1;i<=n;i++) p[i]=i; while(m--) { int a,b; cin>>op>>a>>b; if(op==1) p[find(a)]=find(b); else { if(find(a)==find(b)) cout<<"YES"<<endl; else cout<<"NO"<<endl; } } return 0; }
结论:模板题,核心在于如何计算并套用;