【POJ 1182】食物鏈(並查集)

weixin_34120274發表於2016-07-12

三種動物,A吃B,B吃C,C吃A。那麼用並查集時,還要多一個x和根的關係,吃或者被吃或者同類。合併兩個需要更新和祖先的關係。這個關係可以自己畫一畫。

#include<cstdio>
#define N 50005
struct lian
{
    int fa;
    int re;
} f[N];
int find(int x)
{
    if(x==f[x].fa)return x;
    int fa= f[x].fa;
    f[x].fa=find(fa);
    f[x].re=(f[x].re+f[fa].re)%3;
    return f[x].fa;
}
int ans,n,k;
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; i++)
    {
        f[i].fa=i;
        f[i].re=0;
    }
    for(int i=1; i<=k; i++)
    {
        int d,x,y;
        scanf("%d%d%d" ,&d,&x,&y);
        if(x>n||y>n)
        {
            ans++;
            continue;
        }
        int fx=find(x);
        int fy=find(y);
        if(fx!=fy)
        {
            f[fx].fa=fy;
            f[fx].re=(f[y].re-f[x].re+4-d)%3;
        }
        else
        {
            if(f[x].re!=f[y].re&&d==1)
                ans++;
            else if(f[y].re!=(f[x].re+1)%3&&d==2)
                ans++;
        }
    }
    printf("%d",ans);
}

 

相關文章