动物王国中有三类动物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 Input
100 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
思路:
通过建立一个并查集来实现这三种关系,建一个re数组,0表示与根节点同类,1表示被根节点捕食,2表示捕食根节点。
然后边建立并查集边进行判定。
判定就需要对根节点是否相同进行判定了。
即查找根节点操作。查找的操作不仅仅需要进行路径压缩,而且需要改变re[x]的值。这里有个等式,即re[x]=(re[x]+re[temp])%3.
temp为x的父节点.这个关系可以通过画一个很简单的草图推出。跟向量运算差不多。
如果两个元素的根节点相同,那么就需要对他们进行判定了。
- 如果输入的是1,即两个元素同类,那么只要看看当前两个集合是否与根节点的关系相同就可以。
- 如果输入的是2,即X吃Y,那么这中情况下的反例就是X与Y同类,或Y吃X。同类的话,他们与根节点的关系一定相同。即re[x]==re[y],Y吃X这个推起来有点复杂,这里给出两种情况。(1)X与根节点同类,且Y吃根节点即,re[x]=0,re[y]=2;(2)X吃根节点,根节点吃Y,则Y吃根节点。即 re[x]=2,re[y]=1.通过上面的两种情况发现,re[x]=(re[y]+1)%3.
如果两个子节点的根节点不相同,那么就需要联合操作了。联合的过程其实也是向量的运算。 即re[temp2]=(re[x]+d-re[y]+3)%3;
注意这里的d与不能直接用题目中的D进行替换, 因为这里的1,2与题目中的1,,2的含义不同, 不能直接进行运算, 只需要在原基础上减1即可。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=50005;
int n,k;
int re[maxn];
int a[maxn];
int finds(int x)
{
int temp=a[x];
if(a[x]==x)
return x;
int par=finds(a[x]);
re[x]=(re[x]+re[temp])%3;
a[x]=par;
return par;
}
int main()
{
scanf("%d%d",&n,&k);
for (int i=1;i<=n;i++)
{
re[i]=0;
a[i]=i;
}
int num=0;
for (int i=0;i<k;i++)
{
int t,x,y;
scanf("%d%d%d",&t,&x,&y);
int temp1=finds(x);
int temp2=finds(y);
if(x>n||y>n||(t==2&&x==y))
num++;
else if(temp1==temp2)
{
if(t==1&&re[x]!=re[y])
num++;
if(t==2&&(re[x]==re[y]||re[x]==(re[y]+1)%3))
num++;
}
else if(temp1!=temp2)
{
re[temp2]=(t-1+re[x]-re[y]+3)%3;
a[temp2]=temp1;
}
}
printf("%d\n",num);
return 0;
}