题目:
动物王国中有三类动物 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 句话有的是真的,有的是假的。
当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
- 当前的话与前面的某些真的话冲突,就是假话;
- 当前的话中 X或 Y 比 N 大,就是假话;
- 当前的话表示X 吃 X,就是假话。
你的任务是根据给定的 N 和 K 句话,输出假话的总数。
输入格式
第一行是两个整数 N和 K,以一个空格分隔。
以下 K 行每行是三个正整数 D,X,Y两数之间用一个空格隔开,其中 D 表示说法的种类。
若 D=1,则表示 X和 Y是同类。
若 D=2,则表示 X 吃 Y。
输出格式
只有一个整数,表示假话的数目。
数据范围
1≤N≤50000
0≤K≤100000
输入样例:
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例:
3
分析:
该题的关键在于维护一个数组d[ ],用于表示每个节点到根节点的距离;
假设有一条食物链 a—b—c—d—e—f—g (其中a为根节点,并且b吃a)
那么d[a]=0 d[b]=1 d[c]=2 d[d]=3 d[e]=4 d[f]=5 d[g]=6
由于abcdefg必定为A,B,C中的一类,所以
由b吃a,c吃d,则有a吃c;因为d吃c,所以a,d为同类
同理,a,d,g为一类;b,e为一类;c,f为一类;
根据规律可知
若d[x]%3=0,则x和根节点是同类==>若(d[x]-d[y])%3=0,则x和y是同类
若d[x]%3=1,则x能够吃根节点同类==>若(d[x]-d[y])%3=1,则x能够吃y
若d[x]%3=2,则x会被根节点同类吃==>若(d[x]-d[y])%3=2,则x会被y吃
本题可以分为以下几种情况讨论
一、x>n或者y>n,那么当前的话为假,res++;
二、若x<=n且y<=n,那么
1、t=1时;
(1)若x,y在同一集合,则判断x,y是否为同类
(2)若x,y在不同集合,则将x,y所在集合合并
2、t=2时;
(1)若x,y在同一集合,则判断x是否能吃掉y
(2)若x,y在不同集合,则将x,y所在集合合并
代码如下:
#include<iostream>
using namespace std;
const int N=100010;
int n,m;
int p[N],d[N];//p[N]寻找父节点,d[]求到祖宗节点的距离
int find(int x){ //用于查询x的根节点
if(p[x]!=x) //判断x是否为根节点
{
int t=find(p[x]); //用t暂存p[x]的根节点
d[x]+=d[p[x]]; //d[x]一开始表示x到父节点的距离,d[p[x]]表示p[x]到根节点的距离
p[x]=t;
}
return p[x]; //返回x的根节点
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++)p[i]=i; //表示初始状态时,每个节点的根节点是它本身
int res=0;
while(m--){
int t,x,y;
cin>>t>>x>>y;
if(x>n||y>n)res++;
else{
int px=find(x),py=find(y); //寻找祖宗节点
if(t==1){ //判断是否为同类
if(px==py){ //如果x,y在同一个集合中
if((d[x]-d[y])%3) res++; //两数到根节点距离之差的模不为0,说明不是同类
// 因为d[x]和d[y]可能是负数,C++里负数模正数还是负数。
//所以式子可以写为(d[x] % 3 + 3) % 3 != (d[y] % 3 + 3) % 3
}else{ //如果x,y不在同一个集合中
p[px]=py; //x所在集合并到y所在集合
d[px]=d[y]-d[x];
//由于x->px的距离为d[x]
// y->py的距离为d[y]
//把两个集合合并,可以将px指向py
//那么px->py的距离为d[px] (py为祖宗节点)
//由于x与y同类,所以x与y到根节点py距离之差是3的倍数
//所以(d[x]+d[px]-d[y])%3=0
//所以可以设 d[px]=d[y]-d[x] d[x]在find函数中更新
}
}
if(t==2){
if(px==py){
if((d[x]-1-d[y])%3)res++;
}else{
p[px]=py;
d[px]=d[y]-d[x]+1;
//同理 d[x]+d[px]-d[y]=1
}
}
}
}
cout<<res;
return 0;
}