code1074 食物链

开3*n的并查集,其中x用来连接与x同类的,x+n用来连接x吃的,x+2*n用来连接x被吃的。

1 x y时,如果 x吃y 或 x被y吃,那么为假话,

否则x与y同类,x吃的y也吃,x被吃的y也被吃;

2 x y时,如果 x与y同类(x与x自然也是同类) 或 y吃x,那么为假话,

否则x吃y,y被x吃,y吃x被吃的。

代码:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<queue>
#include<map>
#include<stdlib.h>
using namespace std;

int f[150005];

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

void he(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)return;
    f[y]=x;
}


int main()
{
    int n,m,ans=0;
    cin>>n>>m;

    for (int i=1;i<=3*n;i++)f[i]=i;

    for (int i=1;i<=m;i++)
    {
        int op,x,y;
        cin>>op>>x>>y;
        if (x<1||y<1||x>n||y>n||(x==y&&op==2))ans++;
        else
        {
            if (op==1)
            {
                if (find(x)==find(y+n)||find(x)==find(y+2*n))ans++;
                else
                {
                    he(x,y);
                    he(x+n,y+n);
                    he(x+2*n,y+2*n);
                }
            }
            else
            {
                if (find(x)==find(y)||find(y+2*n)==find(x))ans++;
                else
                {
                    he(x,y+n);
                    he(x+n,y+2*n);
                    he(x+2*n,y);
                }
            }
        }
    }
    cout<<ans;
}
原文地址:https://www.cnblogs.com/FuTaimeng/p/5604927.html