NOI2001_食物链

  • 题目链接:食物链
  • 题目思路:通过题目,我们不难发现有动物之间有三个关系,分别为同类、食物、天敌,于是我们可以开3*n的空间来进行并查集,分别对应上面的关系,进行一些合理的判断,就可以AC。
  • 代码
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    #include<iostream>
    #include<cstdio>
    #include<cstring>
    using namespace std;
    int n,k,ans;
    int f[50010*3];
    int find(int x)
    {
    if(f[x]!=x) f[x]=find(f[x]);
    return f[x];
    }
    void un(int x,int y)
    {
    f[find(y)]=find(x);
    }
    int main()
    {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=3*n;i++)
    f[i]=i;
    for(int i=1;i<=k;i++)
    {
    int c,x,y;
    scanf("%d%d%d",&c,&x,&y);
    if(x>n||y>n) //判断动物序号是否超出范围
    {
    ans++;
    continue;
    }
    if(c==1) //动物为同类
    {
    if(find(x+n)==find(y)||find(x+2*n)==find(y)) //x的食物和y的同类为同类 或 x的天敌和y的同类是同类,显然不符题意
    {
    ans++;
    continue;
    }
    un(x,y); //x,y的同类为同类
    un(x+n,y+n); //x,y的食物合并
    un(x+2*n,y+2*n); //x,y的天敌合并
    }
    else if(c==2) //动物为被吃关系
    {
    if(find(x)==find(y)||find(x+2*n)==find(y)) //x的同类和y的同类为同类 或 x的天敌和y的同类是同类,不符题意
    {
    ans++;
    continue;
    }
    un(x,y+2*n); //x的同类和y的天敌合并
    un(x+n,y); //x的食物和y的同类合并
    un(x+2*n,y+n); //x的天敌和y的食物合并
    }
    }
    printf("%d",ans);
    return 0;
    }
rp++