1182:食物链
总时间限制: 1000ms 内存限制: 65536kB
描述
动物王国中有三类动物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),输出假话的总数。
输入
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
输出
只有一个整数,表示假话的数目。
样例输入
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
样例输出
3
【思路】看了一个大神的思路,非常巧妙,容易理解,大概是给每个动物三个节点,用这三个节点的并查集来确定两动物之间的关系,原文链接:https://blog.csdn.net/lisong_jerry/article/details/80029967,在此简单阐述一下大神的思路
【构建】数组tree[N*3],表示N*3个点的父节点。如下图,动物a、b分别用三个节点表示,a的三个节点的父节点为tree[a] tree[a+N] tree[a+2*N]
1.同类关系的表述:tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N])
2.吃与被吃关系的表述:tree[a]==tree[b+N],tree[a+N]==tree[b+2N],tree[a+2*N]==tree[b])
3.(第一种ab同类关系)a吃c,b吃c:查找root时发现,tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N])
4.(第二种ab同类关系)a吃c c吃d d吃b关系:(查找root时发现,tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N])
【综上所述】
如果ab同类,则必有tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N]),如果a吃b,则必有tree[a]==tree[b+N],tree[a+N]==tree[b+2N],tree[a+2*N]==tree[b]),根据如上关系可以编写代码:
#include<iostream>
#include<cstring>
#include<vector>
#include<math.h>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
#define maxi 0x3f3f3f3f
int N,K;
int tree[200000];// 3*N
int findroot(int x)
{
if(tree[x]==x)
return x;
else
{
int temp=findroot(tree[x]);
tree[x]=temp;
return temp;
}
}
bool same(int x,int y)
{
if(findroot(x)==findroot(y))return true;
return false;
}
void unite(int x,int y)
{
int fx=findroot(x);
int fy=findroot(y);
if(fx!=fy) // 忘掉了,应该是把他们的root连通,而不是让tree[x]=y
tree[fx]=fy;
}
int main(int argc, char const *argv[])
{
cin>>N>>K;
int ans=0;
int d,x,y;
for(int i=1;i<200000;i++)tree[i]=i;
while(K--)
{
cin>>d>>x>>y;
if(x>N||y>N||x<1||y<1)ans++;
else if(d==2&&x==y)ans++;
else if(d==1)
{
if(same(x,y+N)||same(y,x+N))ans++; //如果是x吃y或者y吃x的关系 直接谎话+1
else
{
unite(x,y);
unite(x+N,y+N);
unite(x+2*N,y+2*N);
}
}
else// d==2
{
if(same(x,y)||same(y,x+N))ans++; // 如果是同类或者y吃x 直接谎话+1
else
{
unite(x,y+N);
unite(x+N,y+2*N);
unite(x+2*N,y);
}
}
}
cout<<ans<<endl;
return 0;
}