Hdu 1829 A Bug's Life && Poj 1182 食物链 （并查集偏移量的应用）

复习了一下并查集偏移量的应用，A Bug's Life比食物链简单些，偏移量只有0和1。
偏移量：维护所有集合的根的偏移量都为0。在集合合并时，若两集合对应偏移量不同，改变被合并集合根的偏移量，而集合内部的偏移量暂不改变，在以后查询某点偏移量时，先调用find函数，在find函数中递归更新偏移量值。

Hdu 1829 A Bug's Life

输入可交配的虫子，判断有没有同性交配

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
int p[2005], g[2005], n, flag, root[2005];
void init(){
    memset(g, 0, sizeof(g));
    for(int i=0; i<=n; i++){
        p[i]=i;
        root[i]=0;
    }
    flag=0;
}
int find(int x){
    if(p[x]==x)
        return x;
    int t=find(p[x]);
    g[x]=(g[x]+g[p[x]])%2;
    p[x]=t;
    return p[x];
}
void union_set(int x, int y){
    int x1=find(x), y1=find(y);
    if(root[x1]<root[y1]){
        p[x1]=y1;
        if(g[x]==g[y])g[x1]=1;
    }
    else {
        p[y1]=x1;
        if(g[x]==g[y])g[y1]=1;
    }
    if(root[x1]==root[y1])root[x1]++;
}
int main(){
    //freopen("1.txt", "r", stdin);
    int T, i, x, y, m, cas=1, x1, y1;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &m);
        init();
        while(m--){
            scanf("%d%d", &x, &y);
            if(flag)continue;
            x1=find(x);y1=find(y);
            if(x1==y1){
                if(g[x]==g[y])
                flag=1;
            }
            else {
                union_set(x, y);
            }
        }
        if(flag)
        printf("Scenario #%d:\nSuspicious bugs found!\n\n", cas++);
        else printf("Scenario #%d:\nNo suspicious bugs found!\n\n", cas++);
    }
    return 0;
}

Poj 1182 食物链

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<stack>
#include<queue>
using namespace std;
int p[50005], g[50005], n, k, ans;
void init(){
    memset(g, 0, sizeof(g));
    for(int i=0; i<=n; i++){
        p[i]=i;
    }
    ans=0;
}
int find(int x){
    if(p[x]==x)
        return x;
    int t=find(p[x]);
    g[x]=(g[x]+g[p[x]])%3;
    p[x]=t;
    return p[x];
}
void check(int d,int x,int y){
    int t1=find(x),t2=find(y);
    if(t1==t2){
        if(d==1){
            if(g[x]!=g[y])
            ans++;
        }
    else if((g[y]-g[x]+3)%3!=d)
        ans++;
    return;
    }
    p[t1]=t2;
    if(d==1)
    g[t1]=(g[y]-g[x]+6)%3;
    else g[t1]=(g[y]-g[x]+1+6)%3;
}

int main(){
    //freopen("1.txt", "r", stdin);
    int i, x, y, d;
    scanf("%d%d", &n, &k);
	init();
    while(k--){
        scanf("%d%d%d", &d, &x, &y);
        if(x>n||y>n||(d==2&&x==y)){
        ans++;continue;
        }
        check(d, x, y);
    }
    printf("%d\n", ans);
    return 0;
}