2(銀河英雄伝説)加重互いに素なセットテンプレート

この質問30,000の列があります。

そう右バンド互いに素なセットオリジナルテンプレートマストに基づいて、アレイNUMを追加し、NUM [i]は、キューiの長さを表します。

露出まで可能。

コード:

#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;
    
} 

おすすめ

転載: www.cnblogs.com/ShineEternal/p/11265092.html