原题:https://www.luogu.org/problemnew/show/P2024
题解
并查集能维护连通性、传递性,通俗地说,亲戚的亲戚是亲戚。然而当我们需要维护一些对立关系,比如 敌人的敌人是朋友 时,正常的并查集就很难满足我们的需求。这时,种类并查集就诞生了。常见的做法是将原并查集扩大一倍规模,并划分为两个种类。在同个种类的并查集中合并,和原始的并查集没什么区别,仍然表达他们是朋友这个含义。考虑在不同种类的并查集中合并的意义,其实就表达 他们是敌人 这个含义了。按照并查集美妙的 传递性,我们就能具体知道某两个元素到底是 敌人 还是 朋友 了。至于某个元素到底属于两个种类中的哪一个,由于我们不清楚,因此两个种类我们都试试。
在本题中,用3倍的并查积的存各种动物的关系,一倍存本身,二倍存猎物,三倍存天敌,唯一容易忽略的点就是:一的猎物的猎物就是一的天敌即这是一个循环生态链。那么我们每次只要维护三个并查积的关系就可以了。
#include<cstdio>
int fa[300005];
int n,k,ans;
inline int read()
{
int sum=0;
char ch=getchar();
while(ch>'9'||ch<'0') ch=getchar();
while(ch>='0'&&ch<='9') sum=sum*10+ch-48,ch=getchar();
return sum;
}//读入优化
int find(int x)
{
if(x!=fa[x]) fa[x]=find(fa[x]);
return fa[x];
}//查询
int unity(int x,int y)
{
int r1=find(fa[x]),r2=find(fa[y]);
fa[r1]=r2;
}//合并
int main()
{
int x,y,z;
n=read(),k=read();
for(int i=1;i<=3*n;++i) fa[i]=i; //对于每种生物:设 x 为本身,x+n 为猎物,x+2*n 为天敌
for(int i=1;i<=k;++i)
{
z=read(),x=read(),y=read();
if(x>n||y>n) {ans++; continue;} // 不属于该食物链显然为假
if(z==1)
{
if(find(x+n)==find(y)||find(x+2*n)==find(y)) {ans++; continue;}
//如果1是2的天敌或猎物,显然为谎言
unity(x,y); unity(x+n,y+n); unity(x+2*n,y+2*n);
//如果为真,那么1的同类和2的同类,1的猎物是2的猎物,1的天敌是2的天敌
}
else if(z==2)
{
if(x==y) {ans++; continue;} //其实是废话但是可以稍微省点时间
if(find(x)==find(y)||find(x+2*n)==find(y)) {ans++; continue;}
//如果1是2的同类或猎物,显然为谎言
unity(x,y+2*n); unity(x+n,y); unity(x+2*n,y+n);
//如果为真,那么1的同类是2的天敌,1的猎物是2的同类,1的天敌是2的猎物
}
}
printf("%d\n",ans);
return 0;
}