【POJ】1182 食物链

这是《挑战设计程序竞赛》中的例题。

---

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

题意：中文题面。不赘述。

题解：

代码：

//带权并查集
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 150010;

int f[maxn];
int dep[maxn];//深度

void init(int n){
    for(int i = 0; i <= n ;i++){
        f[i] = i;
        dep[i] = 1;
    }
}

int find(int x){
    if(x == f[x])
        return x;
    return f[x] = find(f[x]); 
} 

void join(int x,int y){
    x = find(x);
    y = find(y);
    if(x == y)    return ;
    if(dep[x] < dep[y]){
        f[x] = y;
    }
    else{
        f[y] = x;
        if(dep[x] == dep[y])
            dep[x]++;
    }
}

bool same(int x,int y){
    return find(x) == find(y);
}

int main(){
    int n,k;
    int d,x,y;
    int ans = 0;
    scanf("%d%d",&n,&k);
    init(n*3);
    for(int i = 0; i < k; i++){
        scanf("%d%d%d",&d,&x,&y);
        if(x <= 0 || x > n || y <= 0 || y > n || ( d == 2 && x == y)){
            ans++;
            continue;
        }
        if(d == 1){
            //xy同类
            if( same(x,y+n) || same(x,y+2*n)){
                ans++;
                continue;
            }else{
                join(x,y);
                join(x+n , y+n);
                join(x + 2*n, y + 2*n);
            }
        }else{
            //x吃y
            if(same(x,y)||same(x,y + 2*n)){
                ans++;
                continue;
            }else{
                join(x,y+n);
                join(x + n, y + 2*n);
                join(x + 2*n,y);
            }
        }
    }
    printf("%d",ans);
    return 0;
}