War - UVa 10158 并查集

并查集模拟社交关系

最新推荐文章于 2022-01-11 17:29:22 发布

原创最新推荐文章于 2022-01-11 17:29:22 发布 · 557 阅读

0 ·

CC 4.0 BY-SA版权

UVa 同时被 2 个专栏收录

218 篇文章

订阅专栏

数据结构

67 篇文章

订阅专栏

本文介绍了一种使用并查集解决社交网络中关系判断的问题。通过并查集的数据结构来模拟朋友和敌人的关系变化，实现快速查找和更新关系状态。

题意：朋友的朋友是朋友，朋友的敌人是敌人，敌人的敌人是朋友。

思路：并查集模拟。

AC代码如下：

#include<cstdio>
#include<cstring>
using namespace std;
int n,p[10010],en[10010];
int find(int x)
{
    return p[x]==x ? x :p[x]=find(p[x]);
}
int main()
{
    int i,j,k,u,v,u2,v2;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
       p[i]=i;
    while(~scanf("%d%d%d",&k,&u,&v) && k>0)
    {
        u++;v++;
        u=find(u);
        v=find(v);
        if(k==1)
        {
            if(en[u]==v)
              printf("-1\n");
            else
            {
                p[v]=u;
                u2=find(en[u]);
                v2=find(en[v]);
                if(u2>0 && v2>0)
                  p[v2]=u2;
                else if(v2>0)
                  u2=v2;
                en[u]=u2;
                en[u2]=u;
            }
        }
        else if(k==2)
        {
            if(u==v)
              printf("-1\n");
            else if(en[u]!=v)
            {
                if(en[u]>0)
                {
                    u2=find(en[u]);
                    p[u2]=v;
                }
                if(en[v]>0)
                {
                    v2=find(en[v]);
                    p[v2]=u;
                }
                en[u]=v;
                en[v]=u;
            }
        }
        else if(k==3)
        {
            if(u==v)
              printf("1\n");
            else
              printf("0\n");
        }
        else if(k==4)
        {
            if(en[u]==v)
              printf("1\n");
            else
              printf("0\n");
        }
    }
}