并查集是一种树形的数据结构。它用于处理一些不交集的合并及查询问题
查找:确定某个元素处于哪个子集;合并:将两个子集合并成一个集合。
先上三个模板
并查集
#include<bits/stdc++.h>
using namespace std;
int f[5005],n,m,a,b,q1,q2,p;
int find(int x){
if(f[x]==x){
return x;
}
return f[x]=find(f[x]);
}
void com(int x,int y){
int xx=find(x);
int yy=find(y);
f[xx]=yy;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
cin>>p;
for(int i=1;i<=n;i++){
f[i]=i;
}
for(int i=1;i<=m;i++){
cin>>a>>b;
com(a,b);
}
for(int i=1;i<=p;i++){
cin>>q1>>q2;
if(find(q1)==find(q2)){
cout<<"Yes"<<'\n';
}
else{
cout<<"No"<<'\n';
}
}
}
点带权并查集
#include<bits/stdc++.h>
using namespace std;
int f[100005],num[100005],n,m,a,b,q1,q2,p;
int find(int x){
if(f[x]==x){
return x;
}
return f[x]=find(f[x]);
}
void com(int x,int y){
int xx=find(x);
int yy=find(y);
f[xx]=yy;
num[yy]+=num[xx];
}
string s;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++){
f[i]=i;
num[i]=1;
}
for(int i=1;i<=m;i++){
cin>>s;
if(s=="C"){
cin>>a>>b;
if(find(a)==find(b)){
continue;
}
else{
com(a,b);
}
}
if(s=="Q1"){
cin>>q1>>q2;
if(find(q1)==find(q2)){
cout<<"Yes"<<'\n';
}
else{
cout<<"No"<<'\n';
}
}
if(s=="Q2"){
cin>>a;
cout<<num[find(a)]<<'\n';
}
}
}
边带权并查集
#include<bits/stdc++.h>
using namespace std;
int f[200005],dis[200005],n,m,a,b,q1,q2,p;
int ans=0x3f3f3f3f;
int find(int x){
if(f[x]==x){
return x;
}
int root=find(f[x]);
dis[x]+=dis[f[x]];
return f[x]=root;
}
void com(int x,int y){
int xx=find(x);
int yy=find(y);
if(xx==yy){
ans=min(ans,dis[y]+dis[x]+1);
}
f[xx]=yy;
dis[x]=dis[y]+1;
}
string s;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
f[i]=i;
}
for(int i=1;i<=n;i++){
cin>>p;
com(i,p);
}
cout<<ans;
}
习题二则
集合
时间限制:1秒 内存限制:128M
题目描述
现在给你一些连续的整数,它们是从A到B的整数。一开始每个整数都属于各自的集合,然后你需要进行一下的操作:
每次选择两个属于不同集合的整数,如果这两个整数拥有大于等于P的公共质因数,那么把它们所在的集合合并。
反复如上操作,直到没有可以合并的集合为止。
现在Caima想知道,最后有多少个集合。
输入描述
输入一行,三个整数A,B,P。
其中,A≤B≤100000;2≤P≤B。
输出描述
一个数,表示最终集合的个数。
样例
输入
10 20 3
输出
7
提示
并查集
有80%的数据B≤1000。
样例解释{10,20,12,15,18},{13},{14},{16},{17},{19},{11}。
解题思路
此题的数据量达到了100000,直接在a到b的范围内遍历求解质因数会超时。因此我们逆向思维一下,枚举质数的倍数,在a-b的范围内将该质数于因数合并,最后记录该范围内集合数量即可。
AC代码
#include<bits/stdc++.h>
using namespace std;
int f[100005],n,m,a,b,cmp;
int prime[100005],st=0;
int find(int x){
if(f[x]==x){
return x;
}
return f[x]=find(f[x]);
}
void com(int x,int y){
int xx=find(x);
int yy=find(y);
f[xx]=yy;
}
void pr(int n){
prime[0]=prime[1]=1;
for(int i=2;i<=b;i++){
if(prime[i]==0){
for(int j=i*2;j<=n;j+=i){
prime[j]=1;
if(prime[i]==0&&i>=cmp&&j>=a){
com(i,j);
}
}
}
}
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>a>>b>>cmp;
for(int i=1;i<=b+5;i++){
f[i]=i;
}
pr(b);
int ans=0;
for(int i=a;i<=b;i++){
if(f[i]==i){
ans++;
}
}
cout<<ans;
}
银河英雄传说
时间限制:3秒 内存限制:128M
题目描述
公元五八○一年,地球居民迁移至金牛座α第二行星,在那里发表银河联邦创立宣言,同年改元为宇宙历元年,并开始向银河系深处拓展。 宇宙历七九九年,银河系的两大军事集团在巴米利恩星域爆发战争。泰山压顶集团派宇宙舰队司令莱因哈特率领十万余艘战舰出征,气吞山河集团点名将杨威利组织麾下三万艘战舰迎敌。 杨威利擅长排兵布阵,巧妙运用各种战术屡次以少胜多,难免恣生骄气。在这次决战中,他将巴米利恩星域战场划分成30000列,每列依次编号为1, 2, …, 30000。之后,他把自己的战舰也依次编号为1, 2, …, 30000,让第i号战舰处于第i列(i = 1, 2, …, 30000),形成“一字长蛇阵”,诱敌深入。这是初始阵形。当进犯之敌到达时,杨威利会多次发布合并指令,将大部分战舰集中在某几列上,实施密集攻击。合并指令为M i j,含义为让第i号战舰所在的整个战舰队列,作为一个整体(头在前尾在后)接至第j号战舰所在的战舰队列的尾部。显然战舰队列是由处于同一列的一个或多个战舰组成的。合并指令的执行结果会使队列增大。 然而,老谋深算的莱因哈特早已在战略上取得了主动。在交战中,他可以通过庞大的情报网络随时监听杨威利的舰队调动指令。 在杨威利发布指令调动舰队的同时,莱因哈特为了及时了解当前杨威利的战舰分布情况,也会发出一些询问指令:C i j。该指令意思是,询问电脑,杨威利的第i号战舰与第j号战舰当前是否在同一列中,如果在同一列中,那么它们之间布置有多少战舰。 作为一个资深的高级程序设计员,你被要求编写程序分析杨威利的指令,以及回答莱因哈特的询问。 最终的决战已经展开,银河的历史又翻过了一页……
输入描述
输入的第一行有一个整数T(1< =T< =500,000),表示总共有T条指令。 以下有T行,每行有一条指令。指令有两种格式: 1.M i j :i和j是两个整数(1< =i , j< =30000),表示指令涉及的战舰编号。该指令是莱因哈特窃听到的杨威利发布的舰队调动指令,并且保证第i号战舰与第j号战舰不在同一列。 2.C i j :i和j是两个整数(1< =i , j< =30000),表示指令涉及的战舰编号。该指令是莱因哈特发布的询问指令。
输出描述
你的程序应当依次对输入的每一条指令进行分析和处理: 如果是杨威利发布的舰队调动指令,则表示舰队排列发生了变化,你的程序要注意到这一点,但是不要输出任何信息; 如果是莱因哈特发布的询问指令,你的程序要输出一行,仅包含一个整数,表示在同一列上,第i号战舰与第j号战舰之间布置的战舰数目。如果第i号战舰与第j号战舰当前不在同一列上,则输出-1。
样例
输入
4 M 2 3 C 1 2 M 2 4 C 4 2
输出
-1 1
提示
解题思路
此题是一道标准的并查集问题,唯一特别的c操作,要求输出两艘战舰之间的战舰数量,如果搜索必定超时,而且失去了并查集的意义,因此我们可以通过以下方法解决:
1.因为战舰排为一排,因此我们只需要知道两艘战舰距离祖先的距离,作差-1即可
编号
1
2 dis=1
3
4
5 dis=4
abs(4-1)-1=2(艘)
2.想要记录与祖先的距离,要用num来更新dis
操作:M 1 3
集合一:
1 num=2,dis=0
2 dis=1
集合二:
3 num=3,dis=0
4 dis=1
5 dis=2
合并后:
3 num=3+2=5(比dis[1]的操作更晚),dis=0
4 dis=1
5 dis=2
1 dis=dis+num[find(3)]=0+3=3
2 dis=dis+dis[fa[2]=4
AC代码
#include<bits/stdc++.h>
using namespace std;
int f[200005],dis[200005],num[100005],n,m,a,b,q1,q2,p;
int ans=0x3f3f3f3f;
int find(int x){
if(f[x]==x){
return x;
}
int root=find(f[x]);
dis[x]+=dis[f[x]];
return f[x]=root;
}
void com(int x,int y){
int xx=find(x);
int yy=find(y);
f[xx]=yy;
dis[xx]=num[yy];
num[yy]+=num[xx];
}
string s;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t;
cin>>t;
for(int i=1;i<=30005;i++){
f[i]=i;
num[i]=1;
}
int x,y;
for(int i=1;i<=t;i++){
cin>>s;
if(s=="M"){
cin>>x>>y;
com(x,y);
}
if(s=="C"){
cin>>x>>y;
if(find(x)!=find(y)){
cout<<-1<<'\n';
}
else{
cout<<abs(dis[x]-dis[y])-1<<'\n';
}
}
}
}