洛谷P2024 [NOI2001]食物链 种类并查集

题目链接:https://www.luogu.com.cn/problem/P2024

这题解法有点类似于洛谷P1525关押罪犯,用到种类并查集,只不过关押罪犯用到两种,而这道题用到三种,分别是同类,天敌与猎物。
我们并查集数组开三倍,1-n代表同类,1+n-2n代表猎物,1+2n-3n代表天敌。
如果x与y是同类,x就不能是y的天敌与猎物,否则就是假话,如果x是y的天敌,x就不能是y的猎物与同类,否则就是假话。如果是真话,我们更新这种关系。
代码入下

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+5;
int fa[maxn*3];//三个种类开三倍
int find(int x)//找祖先
{
    if(x==fa[x])
    return x;
    return fa[x]=find(fa[x]);
}
void unit(int x,int y)//合并
{
    int xx=find(x),yy=find(y);
    fa[xx]=yy;
}
int n,k,x,y,opt;
int main()
{
    scanf("%d %d",&n,&k);
    for(int i=1;i<=3*n;i++)
    fa[i]=i;
    int ans=0;
    for(int i=1;i<=k;i++)
    {
        scanf("%d %d %d",&opt,&x,&y);
        if(x>n||y>n)
        {
            ans++;
            continue;//不满足第二个条件
        }
        if(opt==1)
        {
            if(find(x)==find(y+n)||find(x)==find(y+2*n))
            {
                ans++;
                continue;//如果x与y是同类,x不能与y的猎物与天敌是同类,y也不能与x的猎物与天敌是同类
            }
            unit(x,y),unit(x+n,y+n),unit(x+2*n,y+2*n);
        }
        else
        {
            if(x==y)
            {
                ans++;
                continue;//猎物不能是自己
            }
            if(find(x)==find(y)||find(x)==find(y+n))
            {
                ans++;
                continue;//如果x是y的天敌,x不能与y是同类,x不能是y的猎物
            }
            unit(x+n,y),unit(x,y+2*n),unit(x+2*n,y+n);
        }
    }
    printf("%d\n",ans);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值