poj——1182食物链并查集（提升版）

最新推荐文章于 2021-10-01 14:22:11 发布

快乐撸代码

最新推荐文章于 2021-10-01 14:22:11 发布

阅读量151

点赞数

CC 4.0 BY-SA版权

分类专栏： # 并查集

本文链接：https://blog.youkuaiyun.com/DREAM_yao/article/details/99688541

并查集专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用并查集数据结构解决食物链关系判断问题的方法。通过将动物分为ABC三类，并利用并查集进行同类合并，解决了动物间捕食关系的判断，避免了直接比较带来的复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

因为是中文题，题意就不说了，直接说思路：

我们不知道给的说法中的动物属于A B C哪一类，所以我们可以用不同区间的数字表示这几类动物,这并不影响结果，我们可以用并查集把属于一类的动物放在一块，举个例子：

A 2*n ~ 3*n-1

B n ~ 2*n-1

C 0 ~ n-1

假设 n =10，1和3是同类，那么1和3并到一块，11和13并到一块，21和23并到一块。那么这样既不相互矛盾，又把这三种动物的可能性列举了出来。1捕食4，那么说明1比4高了一个等级，将1和14归并为一类，11和24归并为一类，21和4归并为一类，即B吃C,A吃B,C吃A。

看代码：

#include<stdio.h>
int f[160001];
int getf(int v)
{
    if(f[v]==v)
        return v;
    return f[v]=getf(f[v]);
}
void marge(int u,int v)
{
    int t1=getf(u),t2=getf(v);
    if(t1!=t2)
        f[t2]=t1;
}
bool same(int u,int v)
{
    int t1=getf(u),t2=getf(v);
    if(t1!=t2)
        return 0;
    return 1;
}
int main()
{
//A 2*n ~ 3*n-1

//B n ~ 2*n-1

//C 0~ n-1
    int n,m,sum,a,b,p;
    scanf("%d%d",&n,&m);
    for(int i=0; i<n*3+10; i++)
        f[i]=i;
    sum=0;
    for(int i=0; i<m; i++)
    {
        scanf("%d%d%d",&p,&a,&b);
        a-=1,b-=1;
        if(a>=n||b>=n||b<0||a<0)
        {
            sum++;
            continue;
        }
        if(p==1)
        {
            if(same(a,b+n)||same(a,b+2*n))//a和b不能有捕食和被捕食的关系
                sum++;
            else
            {
                marge(a,b);//同一等级
                marge(a+n,b+n);
                marge(a+2*n,b+2*n);
            }
        }
        else if(p==2)
        {
            if(same(a,b)||same(a,b+2*n))//a和不能是同类 也不能是b捕食a
                sum++;
            else//a捕食b表示a和b+n是同一等级
            {
                marge(a,b+n);
                marge(a+n,b+2*n);
                marge(a+2*n,b);
            }
        }
    }
    printf("%d\n",sum);
    return 0;
}