题目链接: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;
}