扩展域并查集--食物链

并查集用于维护一些有相同相似性质的元素,但有时元素可能有多种状态,比如说一个元素可能是A状态可能是B状态可能是C状态,然后给出一系列诸如“a属于A则b属于B”“a属于B则b属于C”这样的限制条件,那么这时想必一个并查集很难处理,就得用上扩展域并查集,将一个点拆成多个点表示这个点所处的不同状态,就能用并查集维护了

例题:[NOI2001]食物链

链接:https://www.luogu.org/problemnew/show/P2024

一点拆三点,分情况判断。

#include<bits/stdc++.h>
using namespace std;
int n,m,fa[200010];// self eat enemy
int find(int x)
{return fa[x]==x?x:fa[x]=find(fa[x]);}
void mg(int x,int y)
{
    if(find(x)!=find(y))
        fa[find(x)]=find(y);
}
int main()
{
    int ans=0,op,x,y;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=3*n;i++)fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&op,&x,&y);
        if(x>n||y>n){ans++;continue;}
        if(op==1)
        {
            if(find(x+n)==find(y)||find(y+n)==find(x))ans++;
            else mg(x,y),mg(x+n,y+n),mg(x+n+n,y+n+n);
        }
        else 
        {
            if(find(x)==find(y)||find(y+n)==find(x))ans++;
            else mg(x+n,y),mg(x,y+n+n),mg(y+n,x+n+n);
        }
    }
    cout<<ans;
}

猜你喜欢

转载自blog.csdn.net/caoyang1123/article/details/81434880