并查集习题总结

1. 团伙https://www.luogu.com.cn/problem/P1892

如果这两个人是朋友，那么直接合并，即一个人成为另一个人的爸爸

如果这两个人是敌人，用一个数组把每个人的敌人存起来，然后循环找队友，继续认爹

最后只要看有多少人的爹是自己，就有多少对团伙

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
int n,m,prt[1001],dr[1001];
int ans[1001];
int find(int u){
	if(prt[u]==u) return u;
	prt[u]=find(prt[u]);
	return prt[u];
}
void hebing(int x,int y){
    int xx=find(x);
	int yy=find(y); 
    if(xx!=yy) prt[yy]=xx;
    return;
}
void zs(){
    for(int i=1;i<=n;i++) prt[i]=i;	
}
signed main(){
	
    cin>>n>>m;
	zs();
    for(int i=1;i<=m;i++){
        int p,q;
        char c;
        cin>>c>>p>>q;
        if(c=='F') hebing(p,q);
        else{
            if(dr[p]==0) dr[p]=find(q);
            else hebing(q,dr[p]); 
            if(dr[q]==0) dr[q]=find(p);
            else hebing(p,dr[q]);
        }
    }
    for(int i=1;i<=n;i++) ans[find(i)]++;
    int sum=0;
    for(int i=1;i<=n;i++)
    if(ans[i]) sum++; 
    cout<<sum;
    return 0;
} 

2.
关押罪犯​​​​​​https://www.luogu.com.cn/problem/P1525

按照从大到小的顺序排序影响值。从大到小开始”劝架”

若两个人在相同集合，则他们的影响值一定是最大的。

若两个人均有对立集合，则合并x与 y的对立集合，且合并 y与x的对立集合。

若两个人中有任意一人有对立集合，则合并对立集合和没对立集合的罪犯，

并将没对立集合的罪犯的对立集合设为另一罪犯。

若两个人均无对立集合，则将彼此设为对立 集合

  

 参考代码：

#include<iostream>
#include<algorithm>
#pragma GCC optimzie(2)
using namespace std;
int f[200005];
struct node{
	int x,y,z;
}a[100005];
int cmp(node a,node b){
	return a.z>b.z;
}
int find(int x){   
    if(f[x]!=x) f[x]=find(f[x]);
    return f[x];
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    int q;
    for(int i=1;i<=2*n;i++) f[i]=i;
    for(int i=1;i<=m;i++){
    	scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
	}
	sort(a+1,a+m+1,cmp);
	for(int i=1;i<=m;i++){
		int xx=find(a[i].x);
		int yy=find(a[i].y);
		if(xx==yy){
			printf("%d",a[i].z);
			return 0;
		}else{
			f[xx]=find(a[i].y+n);
			f[yy]=find(a[i].x+n);
		}
	}
	printf("0");
	return 0;
}
 

3. 银河英雄传说https://www.luogu.com.cn/problem/P1196

 对于每个点，分别记录所属链的头结点、该点到头结点的距离以及它所在集合的大小。

每次合并将y接在x的尾部，改变y头的权值和所属链的头结点，同时改变x的尾节点。

注意：每次查找的时候也要维护每个节点的权值。

每次查询时计算两点的权值差。

#include<bits/stdc++.h>
using namespace std;
#pragma GCC optimzie(2)
int prt[500005],b[500005],c[500005];
int find(int u){
	int dad;
	if(u==prt[u]) return u; 
	dad=find(prt[u]);
	b[u]=b[u]+b[prt[u]];
	prt[u]=dad;
	return prt[u];
}
int pd(int x,int y){
	if(find(x)!=find(y)){
		return false;
	}
	return true;
}
void ssss(int x,int y){
	x=find(x);
	y=find(y);
	prt[y]=x;
	b[y]=b[y]+c[x];
	c[x]=c[x]+c[y];
}
int main(){
	ios::sync_with_stdio(0);
	int n;
	cin>>n;
	for(int i=1;i<=n;i++) prt[i]=i,c[i]=1;
	for(int i=1;i<=n;i++){
		char d;
		cin>>d;
		if(d=='M'){
			int a,b;
			cin>>a>>b;
			ssss(a,b);
		}
		else{
			int y,x;
			cin>>x>>y;
			if(!pd(x,y)) cout<<"-1"<<endl;
			else{
			cout<<abs(b[x]-b[y])-1<<endl;
			}
		}
	}
	return 0;
}

4.食物链https://www.luogu.com.cn/problem/P2024 

由题意得，动物一共只有 A,B,C 三种，也就是说只要确定了一种动物的种类和他们的关系（即权值），其他的动物的种类也就知道了。

我们用 re[x] 表示编号 i 与父亲节点的权值关系，由于只有三种动物，所以权值也只有三种：

• 0--> 同种动物，
• 1--> 捕食关系，
• 2--> 被捕食关系，

转移时便可以采用对 33 取模来实现。（初始化为 0 ，即自己与自己为同种动物）

#include <bits/stdc++.h>
using namespace std;
const int N=5e4+10;
int n,m;
int pre[N];
int re[N];	
int ans=0;
int getfa(int x) {
	if(pre[x]==x) return x; 
	int fa=pre[x];
	pre[x]=getfa(fa);
	re[x]=(re[x]+re[fa])%3;
	return pre[x];
}
int main() {
	cin>>n>>m;
	for(int i=1; i<=n; i++) pre[i]=i,re[i]=0;
	for(int i=1;i<=m;i++){
		int p,a,b;
		cin>>p>>a>>b;
		if((a>n||b>n)||(p==2&&a==b)){
			ans++;
			continue;
		}
		if(p==1){
			int f1=getfa(a);
			int f2=getfa(b);
			if(f1==f2&&(re[a]-re[b]+3)%3!=0){
				ans++;
				continue;
			} 
			else if(f1!=f2){
				pre[f1]=f2;
				re[f1]=(3-re[a]+re[b])%3;
			}
		}
		if(p==2){
			int f1=getfa(a);
			int f2=getfa(b);
			if(f1==f2){
				int t=(re[a]-re[b]+3)%3;
				if(t!=1){
					ans++;
					continue;
				}
				
			}
			else{
				int f1=getfa(a);
				int f2=getfa(b);
				pre[f1]=f2;
				re[f1]=(re[b]-re[a]+1+3)%3;
			}
		}
	}
	cout<<ans<<endl;
	return 0;
}

 

5.狡猾的商人https://www.luogu.com.cn/problem/P2294 

 给出[l,r]的区间和，相当于sum[r]-sum[l]

一旦已经知道了 sum[a]-sum[b],sum[b]-sum[c]，显然再给出一条sum[a,c]就可以判断"账本的真假"了

将每条这样的信息(l,r,t)，l,r放入一个集合中，用并查集来维护，并维护pre[l]=sum[cnt]-sum[l],pre[r]=sum[cnt]-sum[r]

若 l,r已经在同一个集合中，就直接查询pre[l]-pre[r]，判读与t是否相等

#include<bits/stdc++.h>
using namespace std;
int fa[105],pre[105]; 
int n,m; 
int find(int x){  
	if(x==fa[x]) return x;
    int t=find(fa[x]);
    pre[x]+=pre[fa[x]];
    fa[x]=t;  
    return fa[x];
}  
int main() {
	int t;
    cin>>t;
    while(t--){  
        int flag=0;  
        cin>>n>>m;
        for(int i=0;i<=n;i++){  
            fa[i]=i;  
            pre[i]=0;  
        }  
        for(int i=1;i<=m;i++){  
            int x,y,z;
            cin>>x>>y>>z; 
            x--;  
            if(find(x)!=find(y)){  
                pre[fa[y]]=pre[x]-pre[y]-z;  
                fa[fa[y]]=fa[x];
            }  
            else if(pre[x]-pre[y]!=z) flag=1;  
        }  
        if(flag==0) cout<<"true"<<endl; 
		else cout<<"false"<<endl;  
    }  
    return 0;  
}

完结撒花，感谢支持！！！