食物链（POJ-1182)

经典并查集的使用。

并查集可以高效的维护集合之间的关系。还可以判断图的连通性。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int max_n = 50005*3;
const int max_k = 100005;
int N,K,par[max_n],rankk[max_n];
int T[max_k],X[max_k],Y[max_k];
void init(int n) {
    for(int i=0;i<n;i++) {
        par[i] = i;
        rankk[i] = 0;
    }
}
int findd(int x) {
    return par[x] == x ? x : par[x] = findd(par[x]);
}
bool same(int a,int b) {
    return findd(a) == findd(b);
}
void unite(int x,int y) {
    x = findd(x);
    y = findd(y);
    if(x==y) return ;
    if(rankk[x]<rankk[y]) {
        par[x] = y;
    } else {
        par[y] = x;
        if(rankk[x] == rankk[y]) rankk[x]++;
    }
}
void solve() {
    init(N*3);
    int ans = 0;
    for(int i=0;i<K;i++) {
        int t = T[i];
        int x = X[i] - 1,y = Y[i] - 1;
        if(x<0||N<=x||y<0||N<=y) {
            ans++;
            continue;
        }
        if(t == 1) {
            if(same(x,y+N)||same(x,y+2*N)) {
                ans++;
            }
            else {
                unite(x,y);
                unite(x+N,y+N);
                unite(x+N*2,y+N*2);
            }
        }
        else {
            if(same(x,y) || same(x,y+2*N)) {
                ans++;
            }
            else {
                unite(x,y+N);
                unite(x+N,y+2*N);
                unite(x+2*N,y);
            }
        }
    }
    printf("%d\n",ans);
}
int main() {
    scanf("%d%d",&N,&K);
    for(int i=0;i<K;i++) {
        scanf("%d",&T[i]);
        scanf("%d%d",&X[i],&Y[i]);
    }
    solve();
    return 0;
}