code1074 食物链

Posted 2020-07-28 FuTaimeng

开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;
}