POJ 1182 食物链（并查集拆点）

 

【题目链接】 http://poj.org/problem?id=1182

 

【题目大意】

　　草原上有三种物种，分别为A,B,C

　　A吃B，B吃C，C吃A。

　　1 x y表示x和y是同类，2 x y表示x吃y

　　问给出的信息有几条是和前面相违背的

 

【题解】

　　之前这道题是用加权并查集做的，做的有些晕晕乎乎，现在换了种思路做就清晰很多了

　　将每个点拆点，比如x拆为，x-A，x-B，x-C

　　表示x属于A类，x属于B类，和x属于C类，

　　如果y和x属于同类，那么合并x-A和y-A，x-B和y-B，x-C和y-C

　　如果是x吃y的情况，那么合并x-A和y-B，x-B和y-C，x-C和y-A

　　也就是说将x属于A这种类型的条件作为节点进行并查集运算

　　就能很轻松地解决这种问题

 

【代码】

#include <cstdio>
const int N=50010*3;
int f[N],n,k,ans=0,op,x,y;
void init(int n){for(int i=0;i<=n;i++)f[i]=i;}
int sf(int x){return f[x]==x?x:f[x]=sf(f[x]);}
bool Same(int x,int y){return sf(x)==sf(y);}
bool Union(int x,int y){f[sf(x)]=sf(y);}
int main(){
    scanf("%d%d",&n,&k);
    init(3*n);
    for(int i=1;i<=k;i++){
        scanf("%d%d%d",&op,&x,&y);
        if(--x<0||x>=n||--y<0||y>=n){ans++;continue;}
        if(op==1){
            if(Same(x,y+n)||Same(x,y+2*n))ans++;
            else{Union(x,y);Union(x+n,y+n);Union(x+2*n,y+2*n);}
        }else{
            if(Same(x,y)||Same(x,y+2*n))ans++;
            else{Union(x,y+n);Union(x+n,y+2*n);Union(x+2*n,y);}
        }
    }printf("%d\n",ans);
    return 0;
}