食物链
动物王国中有三类动物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
并查集:
通过树形结构建立集合元素间的从属关系。
此题的fa数组长度声明为maxn的三倍。由于给出的关系中没有指明元素具体是属于A、B或C,但题目给出只有A、B、C这三种动物,所以将fa数组长度定义为三倍,然后将每3种情况模拟出来。
例如:1 50 51
假设50是A,那51就是B;如果假设50是B,那51就是C;如果50是C,就假设51是A。
代码:
#include<iostream>
#include<cstdio>
const int maxn=5e6+5;
int fa[3*maxn],rank[maxn],D[maxn];
int X[maxn],Y[maxn];
int n,k;
void init(int num){//常用操作前面解释过了
for(int i=1;i<=num;i++){
fa[i]=i;
rank[i]=0;
}
}
int find(int x){
if(fa[x]==x)return x;
else return fa[x]=find(fa[x]);
}
void unite(int x,int y){
int xx=find(x);
int yy=find(y);
if(rank[xx]<rank[yy]){
fa[xx]=yy;
}else{
fa[yy]=xx;
if(rank[xx]==rank[yy])rank[xx]++;
}
}
bool same(int x,int y){
return find(x)==find(y);
}
int main(){
scanf("%d %d",&n,&k);
init(3*n);
for(int i=1;i<=k;i++){
scanf("%d %d %d",D+i,X+i,Y+i);
}
//扫描
int ans=0;
for(int i=1;i<=k;i++){
int xx=X[i],yy=Y[i];
if(xx>n||xx<1||yy>n||yy<1){
ans++;
continue;
}
if(D[i]==1){
if(same(xx,yy+n)||same(xx,yy+2*n))ans++;//判断是不是同一类,也就是不是在一个区
else {
unite(xx,yy);//在A区合并
unite(xx+n,yy+n);//在B区合并
unite(xx+2*n,yy+2*n);//在C区合并
}
}else{
if(same(xx,yy)||same(xx,yy+2*n))ans++;//判断谁吃谁
else{
unite(xx,yy+n);//谁吃谁
unite(xx+n,yy+2*n);
unite(xx+2*n,yy);
}
}
}
printf("%d\n",ans);
return 0;
}