P2024 [NOI2001] 食物链

合集 - 并查集(7)

1.P5836 [USACO19DEC] Milk Visits S2024-04-022.[ABC238E] Range Sums2024-06-033.P8779 [蓝桥杯 2022 省 A] 推导部分和2024-06-034.D. X(or)-mas Tree2024-07-165.P2294 [HNOI2005] 狡猾的商人2024-07-23

6.P2024 [NOI2001] 食物链2024-07-27

7.P1967 [NOIP2013 提高组] 货车运输2024-07-31

收起

原题链接

题解

关系具有矢量特性，因此可以带权并查集维护

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;

int fa[50006];
int val[50006];
int finds(int now)
{
    if(now==fa[now]) return now;

    int tem=fa[now];

    fa[now]=finds(fa[now]);
    val[now]=(val[now]+val[tem])%3;

    return fa[now];
}


void solve()
{
    int n,k;
    cin>>n>>k;

    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
        val[i]=0;
    }

    int ans=0;
    for(int i=1;i<=k;i++)
    {
        int op,x,y;

        cin>>op>>x>>y;

        if(x>n||y>n||op==2&&x==y)
        {
            ans++;
            continue;
        }
        if(op==1)
        {
            int fx=finds(x),fy=finds(y);
            if(fx==fy)
            {
                if(val[x]!=val[y]) ans++;
                continue;
            }
            int tem=val[x];
            fa[fx]=fy;
            val[fx]=(3-tem+val[y])%3;
        }
        else
        {
            int fx=finds(x),fy=finds(y);
            if(fx==fy)
            {
                if((val[x]-val[y]+3)%3!=1) ans++;
                continue;
            }
            int tem=val[x];
            fa[fx]=fy;
            val[fx]=(3-tem+val[y]+1)%3;
        }
    }

    cout<<ans;
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    //cin>>t;
    while(t--) solve();
    return 0;
}