智算之道初赛第二场--情报站

并查集题目,有两个小细节

  • type2的时候输入的x和y的一样的,那么就是知道了某个数的2倍,所以可以当作type1的情况来求解
  • type2输入的x和y来自同一个并查集,那么就不要mix他们,不然qua(并查集中元素的数量)会出错。
/*
 * @Description: 
 * @Autor: Kadia
 * @Date: 2020-07-19 19:43:47
 * @LastEditors: Kadia
 * @Connect: vx:ccz1354 qq:544692713
 * @LastEditTime: 2020-07-19 21:50:45
 */ 
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
int par[1000005];
int qua[1000005];
int vis[1000005];
int findroot(int x)
{
    if(par[x]==x)
        return x;
    else
        return par[x]=findroot(par[x]);
}
void mix(int x,int y)
{
    if(findroot(x)==findroot(y))
        return ;
    if(vis[findroot(x)]||vis[findroot(y)])
        vis[findroot(x)]=vis[findroot(y)]=1;
    qua[findroot(y)]+=qua[findroot(x)];
    par[findroot(x)]=findroot(y);
}
int same(int x,int y)
{
    return findroot(x)==findroot(y);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        qua[i]=1;
        par[i]=i;
        vis[i]=0;
    }
    int op;
    int cnt=0;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&op);
        if(op==1)
        {
            int x;
            scanf("%d",&x);
            if(!vis[findroot(x)])
            {
                cnt+=qua[findroot(x)];
                vis[findroot(x)]=1;
            }
        }
        else
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(a==b)
            {
                int x=a;
                if(!vis[findroot(x)])
                {
                    cnt+=qua[findroot(x)];
                    vis[findroot(x)]=1;
                }
            }
            else
            {
                if(vis[findroot(a)]!=vis[findroot(b)])
                    cnt+=(vis[findroot(a)]==0?qua[findroot(a)]:qua[findroot(b)]);
                mix(a,b);
            }
        }
        printf("%d\n",cnt);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值