题目描述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 Description
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数D,X,Y,两数之间用一个空格隔开,其中 D 表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
输出描述Output Description
只有一个整数,表示假话的数目。
样例输入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
数据范围及提示Data Size & Hint
输入文件
对7句话的分析 100 7
1 101 1 假话
2 1 2 真话
2 2 3 真话
2 3 3 假话
1 1 3 假话
2 3 1 真话
1 5 5 真话
NOI 2001 食物链(eat)
和上一道题类似,主要贴一下自己的一些想法:
A吃B,B吃C,C吃A,如果用x表示A,x+n表示B,x+2n表示C。那么当告诉1 1 3时,动物1和动物3是同类,可能的情况是1,3都是A,1,3都是B或者1,3都是C,所以要merge( 1,3 ),merge(1+n,3+n),merge(1+2n,3+2n)。
当告诉2 2 3时,动物2吃动物3,可能的情况是2是A,B or C,所以:merge(2,3+n),merge(2+n,3+2n),merge(2+2n,3)。当i和j+n在一个集合中时,说明动物i要吃动物j;当i+2n和j在同一个集合时,同样说明i要吃j。以下附上这种思路的代码:
#include<cstdio>
int n,k;
int f[50000*3+10];
int getroot(int x) { return f[x]==x ? x : f[x]=getroot(f[x]); }
inline void merge(int x,int y) { f[getroot(x)]=getroot(y); }
int main()
{
freopen("input.txt","r",stdin);
//freopen("1074.out","w",stdout);
scanf("%d%d",&n,&k);
for(int i=1;i<=3*n;i++) f[i]=i;
int res=0;
while(k--)
{
int d,x,y;
scanf("%d%d%d",&d,&x,&y);
if(x>n||y>n) res++;
else if(d==2 && x==y) res++;
else if(d==1)//the same
{
if(getroot(x)==getroot(y+n)||getroot(x+n)==getroot(y+n+n)||getroot(x+n+n)==getroot(y)) res++;//x eat y
else if(getroot(x+n)==getroot(y)||getroot(x+n+n)==getroot(y+n)||getroot(x)==getroot(y+n+n)) res++;//y eat x
else{
merge(x,y);
merge(x+n,y+n);
merge(x+n+n,y+n+n);
}
}
else{//x eat y
if(getroot(x)==getroot(y)||getroot(x+n)==getroot(y+n)||getroot(x+n+n)==getroot(y+n+n)) res++;//x y are the same
else if(getroot(x+n)==getroot(y)||getroot(x+n+n)==getroot(y+n)||getroot(x)==getroot(y+n+n)) res++;//x eat y
else{
merge(x,y+n);
merge(x+n,y+n+n);
merge(x+n+n,y);
}
}
}
printf("%d\n",res);
return 0;
}
另外,上面代码可以精简,因为:
else{
merge(x,y);
merge(x+n,y+n);
merge(x+n+n,y+n+n);
}
另外一种喜闻乐见的思路是:不使用偏移向量法枚举,而是多用一个数组kind维护0,1,2三个值。当x吃y时,kind[x]比kind[y]大1,同时father[y]充当father[x]的father,即father[father[x]]=father[y];当x与y同类时,kind[x]比kind[y]大0,同样要维护成同一个集合,即father[father[x]]=father[y]。这种思路需要注意的问题是如何维护压缩路径时father的修改,如以下代码所示:
int find(int x){
if(x!=father[x]){
int tmp=father[x];
father[x]=find(father[x]);
kind[x]=(kind[x]+kind[tmp])%3;//顺序不能写反
}
return father[x];
}
由于kind[x]总要比
kind[father[x]]大kind[x],因此发生路径压缩时,相当于更换了father[x],因此要更新kind[x],即
kind[x]=(kind[x]+kind[tmp])%3
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
#define MAXN 50001
using namespace std;
int n,m,p,x,y;
int father[MAXN];
int kind[MAXN];
int cnt;
int find(int x){
if(x!=father[x]){
int tmp=father[x];
father[x]=find(father[x]);
kind[x]=(kind[x]+kind[tmp])%3;//顺序不能写反
}
return father[x];
}
int main(){
//freopen("F:\\练习\\test\\test\\input.txt","r",stdin);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) father[i]=i;
while(m--){
int d,x,y;
scanf("%d%d%d",&d,&x,&y);
if(x>n||y>n){cnt++;continue;}
if(d==1){
if( find(x)==find(y) )
if( kind[x]!=kind[y] ) cnt++;
else;
else{
kind[ father[x] ]=( kind[y]-kind[x]+3 )%3;
father[father[x]]=father[y];
}
}
else{
if(x==y){cnt++;continue;}
if( find(x)==find(y) )
if( kind[x]!=(kind[y]+1)%3 ) cnt++;
else;
else{
kind[father[x]]=(kind[y]-kind[x]+4)%3;
father[father[x]]=father[y];
}
}
}
printf("%d\n",cnt);
return 0;
}