poj 1182 食物链 //并查集http://poj.org/problem?id=1182
食物链
Time Limit: 1000MS Memory Limit: 10000K Total Submissions: 99085 Accepted: 29983 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
第一行是两个整数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
第一种方法,建立父节点,https://blog.csdn.net/freezhanacmore/article/details/8767413
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<climits>
using namespace std;
#define mod 100000007
#define LL long long
int n,k;
int g=1;
int pre[50005];
int r[50005]; //与父节点的关系,0同类,1被父节点吃,2吃父节点
void set(int n){ //初始化
for(int i=1;i<=n;i++){
pre[i]=i; //开始自己是自己的父亲节点
r[i]=0;//开始自己就是自己的父亲,每一个点均独立
}
}
int found(int x){
if(x==pre[x]) return x;
int t=pre[x];
pre[x]=found(pre[x]); //递归找根节点
r[x]=(r[x]+r[t])%3; //递归更新路径上的节点和跟节点的关系
return pre[x];
}
void join(int x,int y,int d){
int fx=found(x),fy=found(y); //找到根节点
pre[fy]=fx; //将y合并到x
r[fy]=(r[x]-r[y]+d-1+3)%3; //按照上述合并方式找出y的根节点和x的根节点的关系
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++) pre[i]=i;
int ans=0;
while(k--){
int d,x,y;
scanf("%d%d%d",&d,&x,&y);
if(x>n||x<1||y>n||y<1){
ans++;continue;
}
else if(found(x)==found(y)){ //如果在同一个树
if(d==1&&r[x]!=r[y]) ans++; //如果和父节点的关系不一样
if(d==2&&(r[x]+1)%3!=r[y])ans++; //如果不是x吃y的关系
}
else {
join(x,y,d); //将两个集合合并
}
}
cout<<ans;
return 0;
}
第二种方法:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<climits>
using namespace std;
#define mod 100000007
#define LL long long
int n,k;
int pre[50005*3];//x表示x属于A。 x+n表示x属于B。 x+2*n表示x属于C。
int found(int x){
if(x==pre[x]) return x;
pre[x]=found(pre[x]);
return pre[x];
}
void joint(int x,int y){
int fx=found(x),fy=found(y);
pre[fy]=fx;
}
int main(){
cin>>n>>k;
for(int i=1;i<=n*3;i++) pre[i]=i; //自己和自己是一组
int ans=0;
while(k--){
int d,x,y;
scanf("%d%d%d",&d,&x,&y);
if(x>n||x<1||y>n||y<1){
ans++;continue;
}
if(d==1){// x是A,那么 y也只能是A
if(found(x)==found(y+n)||found(x)==found(y+2*n))ans++;
else {
joint(x,y);
joint(x+n,y+n);
joint(x+2*n,y+2*n);
}
}
else { //x只能吃y
if(found(x)==found(y)||found(x)==found(y+2*n))ans++;
else {
joint(x,y+n);
joint(x+n,y+2*n);
joint(x+2*n,y);
}
}
}
cout<<ans;
return 0;
}