Y1课程笔记:第九课时(并查集问题)

文章介绍了并查集这种数据结构,包括基本的查找和合并操作,并提供了C++实现的模板。此外,文章还展示了点带权并查集和边带权并查集的应用,以及如何处理询问操作。其中,点带权并查集处理了合并和查询,边带权并查集涉及到路径压缩和权值计算。文章还给出了两个具体的问题实例,一个是关于集合合并的,另一个是关于战舰排列的,并提供了相应的解题策略和AC代码。
摘要由CSDN通过智能技术生成

并查集是一种树形的数据结构。它用于处理一些不交集的合并及查询问题

查找:确定某个元素处于哪个子集;合并:将两个子集合并成一个集合。

先上三个模板

并查集

#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';
	        }
	    }
	}
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值