并查集：习题增补

1.连通块中点的数量

https://www.acwing.com/problem/content/839/
这里追踪了每个连通块的大小

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10;
int f[N],w[N];
int find(int x){
    while(f[x]!=x)x=f[x];
    return f[x];
}//与模板中的不同，这是循环的写法
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)f[i]=i,w[i]=1;
    while(m--){
        string op;cin>>op;
        if(op=="C"){
            int a,b;cin>>a>>b;
            if(find(a)!=find(b)){
                w[find(b)]+=w[find(a)];//加权值的操作，由于最后找的是根上的权值，所以不减去也没关系
                f[find(a)]=find(b);
            }
        }else if(op=="Q1"){
            int a,b;cin>>a>>b;
            if(find(a)!=find(b))cout<<"No"<<endl;
            else cout<<"Yes"<<endl;
        }else{
            int a;cin>>a;
            cout<<w[find(a)]<<endl;
        }
    }
    return 0;
}

2.食物链

https://www.acwing.com/problem/content/242/

拆点并查集

并查集(拓展域)
这道题目我们主要是要开三个拓展的域,也就是天敌域,同类域,以及捕食域.
如果x,y是同类,但是x的捕食域有y,那么错误
如果x,y是同类,但是x的天敌域有y,那么错误
如果x是y的天敌,但是x的同类域中有y,那么错误
如果x是y的天敌,但是x的天敌域中有y,那么错误

//这里我们将三个域,分别转化为了n,n+n,n+n+n.因为读入方面特别烦.
#include <bits/stdc++.h>
using namespace std;
int fa[200000];
int n,m,k,x,y,ans;
int get(int x)
{
    if(x==fa[x]) 
        return x;
    return fa[x]=get(fa[x]);
}
void merge(int x,int y)
{
    fa[get(x)]=get(y);
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=3*n;i++) 
        fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&k,&x,&y);
        if(x>n || y>n) 
            ans++;
        else if(k==1)
        {
            if(get(x)==get(y+n) || get(x)==get(y+n+n)) //如果x,y是同类,但是x是y的捕食中的动物,或者x是y天敌中的动物,那么错误.
                ans++;
            else
            {
                merge(x,y);
                merge(x+n,y+n);
                merge(x+n+n,y+n+n);
            }
        }
        else
        {
            if(x==y || get(x)==get(y) || get(x)==get(y+n)) //x就是y,或者他们是同类,再或者是y的同类中有x
                ans++;//都是假话
            else
            {
                merge(x,y+n+n);//y的天敌域中加入x
                merge(x+n,y);//y的同类域归入x的捕食域
                merge(x+n+n,y+n);//y的捕食域中加入x的天敌域
            }
        }
    }
    cout<<ans<<endl;
}
//x是同类域.
//x+n是捕食域
//x+n+n是天敌域

作者：秦淮岸灯火阑珊已退役
链接：https://www.acwing.com/solution/content/1007/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

带权并查集

https://www.acwing.com/solution/content/1357/

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e4 + 233;
int fa[maxn], d[maxn];
int ff(int x)
{
    if(fa[x] == x) return x;
    int r = ff(fa[x]);
    d[x] += d[fa[x]];
    return fa[x] = r;
}
int main()
{
    int n,k; cin >> n >> k;
    for(int i = 0; i <= n; i++) fa[i] = i;
    int ans = 0;
    for(int i = 1; i <= k; i++)
    {
        int t, a, b;
        scanf("%d%d%d", &t, &a, &b);
        if(a > n || b > n) {ans ++; continue;}
        else if(t == 2 && a == b) {ans++; continue;}
        else
        {
            int rel;
            if(t == 2) rel = 1;
            else rel = 0;
            int x = ff(a), y = ff(b);
            if(x == y) 
            {
                if((((d[a] - d[b]) % 3) + 3) % 3 != rel)
                ans++;
            }
            else
            {
                fa[x] = y;
                d[x] = d[b] - (d[a] - rel);
            }
        }
    }
    cout<< ans;
}

作者：这个显卡不太冷
链接：https://www.acwing.com/solution/content/1357/
来源：AcWing
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。