带权并查集模板2(银河英雄传说)

这道题有30000个列。。。

所以说带权并查集在原模板的基础上就必须再添一个num数组,num[i]表示i队列的长度

算是涨了见识了。。。

code:

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
int fa[30005],val[30005],num[30005];
int find(int x)
{
    if(fa[x]!=x)
    {
        int t=fa[x];
        fa[x]=find(fa[x]);
        val[x]+=val[t]; 
    }
    return fa[x];
}
int main()
{
    int T;
    scanf("%d",&T);
    for(int i=1;i<=30000;i++)
    {
        fa[i]=i;
        num[i]=1;
    }
    char pd;
    int x,y;
    while(T--)
    {
        cin>>pd;
        scanf("%d%d",&x,&y);
        if(pd=='M')
        {
            int px=find(x);
            int py=find(y);
            if(px!=py)
            {
            fa[px]=py;
            val[px]+=num[py];
            num[py]+=num[px];
            num[px]=0;
            }
            //printf("qwq\n");
        }
        else
        if(pd=='C')
        {
            int px=find(x);
            int py=find(y);
            if(px!=py)
            {
                printf("-1\n");
            }
            else
            {
                int tmp;
                tmp=abs(val[y]-val[x]);
                printf("%d\n",tmp-1);
            }
            
        }
    }
    return 0;
    
} 
全部评论

相关推荐

jack_miller:杜:你不用我那你就用我的美赞臣
点赞 评论 收藏
分享
11-03 14:38
重庆大学 Java
AAA求offer教程:我手都抬起来了又揣裤兜了
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务