Description动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。Output只有一个整数,表示假话的数目。Sample Input100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
Sample Output
3
——————————————————————————————————————————————————————————————————————
并查集的一道题,因为只有三种生物也不是很复杂,主要思路用递归在查询父的时候更新关系
WA原因:第一次wa是poj编译器不支持<bits/stdc++.h>。。。第二次是用std::cin进行输入时TLE了,卡了一天换各种算法也没解决,换成scanf后270ms(这得多少数据呀)
——————————————————————————————————————————————————————————————————————
father中存它的父节点(集合中第一个有效输入的x)
relation中存与父节点的关系,0为同类,1为被父吃,2为吃父 (原因是可以根据关系加减后mod3来更新关系)
int father[nMax];
int relation[nMax];//与总父关系,0为同类,1为被父吃,2为吃父
int findF(int son)
{
if(father[son]==son)
{
return son;
}
int key=findF(father[son]);
relation[son]=(relation[father[son]]+relation[son])%3;//关系更新
//son的父节点的父节点已经更新为当前总父,然后son父节点与总父的关系加上son与父节点的关系mod3得son与总父关系
father[son]=key;//把son的父节点也更新为总父
return key;
}
如果x与y的父节点不一致,也就是x和y所属集合没有交集(有交集的话应该已经合并),不可能出现错误,直接更新合并就好了
注意:合并只需要把y总父节点的父节点指向x总父,更新两者关系,其余子节点更新交给下一次寻父
if(xf!=yf)
{
father[yf]=xf;
if(num==1)
{
relation[yf]=(relation[x]+3-relation[y])%3;
}
if(num==2)
{
relation[yf]=(relation[x]+4-relation[y])%3;
}
continue;
}
xy父节点相等时进行错误判定
if(xf==yf)
{
if(num==1&&relation[x]!=relation[y])
{
++wrong;
continue;
}
if(num==2&&relation[y]!=(relation[x]+1)%3)
{
++wrong;
continue;
}
continue;
}
以下是AC代码
#include<iostream>
#define nMax 50010
using namespace std;
int n,k,wrong=0;
int father[nMax];
int relation[nMax];//与总父关系,0为同类,1为被父吃,2为吃父
int findF(int son)
{
if(father[son]==son)
{
return son;
}
int key=findF(father[son]);
relation[son]=(relation[father[son]]+relation[son])%3;//关系更新
father[son]=key;
return key;
}
int main()
{
cin>>n>>k;
for(int i=0;i<=n;++i)
{
father[i]=i;
relation[i]=0;
}
for(int i=0;i<k;++i)
{
int num,x,y;
scanf("%d %d %d",&num,&x,&y);
if(x>n||y>n)
{
++wrong;
continue;
}
if(num==2&&x==y)
{
++wrong;
continue;
}
int xf=findF(x);
int yf=findF(y);
if(xf==yf)
{
if(num==1&&relation[x]!=relation[y])
{
++wrong;
continue;
}
if(num==2&&relation[y]!=(relation[x]+1)%3)
{
++wrong;
continue;
}
continue;
}
if(xf!=yf)
{
father[yf]=xf;
if(num==1)
{
relation[yf]=(relation[x]+3-relation[y])%3;
}
if(num==2)
{
relation[yf]=(relation[x]+4-relation[y])%3;
}
continue;
}
}
printf("%d",wrong);
return 0;
}