知识点
并查集的作用
并查集是一种集并集查询于一体的数据结构,它的时间复杂度是 O ( α ( n ) ) O(\alpha(n)) O(α(n)),阿尔法函数是阿克曼函数的反函数,返回值一般不超过3,数很大时一般不超过7,因此并查集的均摊时间复杂度接近O(1)。
一般用来判断两个点是否相通,也可以用来判环,当相通的两个点连接时就会出现环,环的长度是两个点分别到根结点的路径之和加一。
查询
查询是用来找根节点的,一直往上找父亲,直到找到头就结束,返回根节点,此时便可以进行路径压缩,将这个点的父亲直接变成根节点。
代码:
long long Find(long long x){
if(f[x]==x){
return x;
}
return f[x]=Find(f[x]);
}
并集
是指将两个来自不同集合的节点进行连接,也就是将两个集合合并成一个集合,直接找到对应根节点,将根节点相连即可
代码:
void Union(long long x,long long y){
int fx=Find(x),fy=Find(y);
if(fx==fy){
return;
}
f[fx]=fy;
}
带权并查集
有点带权与边带权两种。
点带权一般是存到根节点的距离
查询时将当前深度加上父节点深度。
并集时只修改当前两个节点即可。
代码:
long long Find(long long x){
if(f[x]==x){
return x;
}
int root=Find(f[x]);
dis[x]+=dis[f[x]];
return f[x]=root;
}
void Union(long long x,long long y){
int fx=Find(x),fy=Find(y);
if(fx==fy){
return;
}
f[fx]=fy;
dis[x]=dis[y]+1;
}
分类并查集
分类并查集是一种专门用来维护复杂关系的并查集。复杂关系如:敌人的敌人是朋友,朋友的朋友是朋友和猎物的天敌是同类等。主要是根据类别数量划分出各个区域来表示一种类别,如果让两个点相连表示这两个点是同一类的,如果以第一种关系为例子,则可以用x表示x及其朋友,用x+n表示x的敌人,连接xy两点,表示xy是朋友关系。
例题
题目:选学霸
题目描述
老师想从n名学生中选m人当学霸,但有k对人实力相当,如果实力相当的人中,一部分被选上,另一部分没有,同学们就会抗议。所以老师想请你帮他求出他该选多少学霸,才能既不让同学们抗议,又与原来的m尽可能接近。
输入样例
4 3 2
1 2
3 4
输出样例
2
数据描述
1 ≤ n , m ≤ 2 ∗ 1 0 4 1≤n,m≤2∗10^4 1≤n,m≤2∗104
选学霸 题解
思路
这道题用带权并查集和01背包做。我们可以将当前节点为根节点时树的元素个数当做点权,每次合并时将当父亲节点点权加上另一节点点权。然后再遍历每个节点,遇到根节点,将它的点权存起来,作为01背包物品的价值和代价,因为人数可以超过,所以背包容量要扩大一倍,然后遍历所有情况,将与原人数差值的绝对值最小的作为答案输出。
代码
#include <algorithm>
#include <cstdio>
#include <iostream>
#include <string>
using namespace std;
int f[20005],num[20005],v[20005],dp[40005],n,m,p,cnt=0,ans,minn=0x3f3f3f3f;
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
num[i]=1;
}
}
int Find(int x){
if(f[x]==x){
return x;
}
return f[x]=Find(f[x]);
}
void Union(int x,int y){
x=Find(x);
y=Find(y);
if(x!=y){
f[x]=y;
num[y]=num[x]+num[y];
}
}
int main() {
scanf("%d %d %d",&n,&m,&p);
init();
for(int i=1;i<=p;i++){
int x,y;
scanf("%d %d",&x,&y);
Union(x,y);
}
for(int i=1;i<=n;i++){
if(f[i]==i){
v[++cnt]=num[i];
}
}
for(int i=1;i<=cnt;i++){
for(int j=m*2;j>=v[i];j--){
dp[j]=max(dp[j],dp[j-v[i]]+v[i]);
}
}
for(int i=1;i<=2*m;i++){
if(abs(dp[i]-m)<minn){
minn=abs(dp[i]-m);
ans=dp[i];
}
}
printf("%d",ans);
return 0;
}