[JZOJ5888]GCD生成树

[JZOJ5888]GCD生成树

题目链接

gugugu

分析

对于N很小的情况,暴力Prim即可

对于值域很小的情况,我的想法与solution不太一样,将值相同的缩成一个点,\(O(w^2)\)预处理出每个数对的\(gcd\)

然后设立两个点集,一个加入生成树集合中的,另一个没有,一开始将1放在1集合,然后遍历值域,将存在的数与1构成的数对的gcd都存在一个\(multiset\)中,然后选最大的.加入一个数后又重复以上操作.时间复杂度似乎是\(O(w^2 log w)\)

正解就比较妙了(还是因为我太菜了)

一个显然的性质就是我们要尽量地选\((w,k \times w)\)这样的数对

于是我们从大到小钦定这个\(w\),枚举它的倍数

但是发现有种情况就是\(k_1\)和\(k_2\)不互质,这样的话看起来选的数对就是错的了,因为它们gcd显然比\(w\)更大

然而我们可以使用并查集维护联通块来避免这种情况

对值域建立联通块,我们每将一对数\((k_1 \times w, k_2 \times w)\)算进答案就将\(k_1 \times w\)和\(k_2 \times w\)所在联通块合并

这样的话就算你\(k_1,k_2\)不互质,由于我们从大到小枚举这个\(w\),这种情况已经是被算过的,你只要每次判断两个数是否在一个联通块内即可去掉这种情况

计算答案就很简单了,你枚举\(w\)倍数时加入了\(k\)对数,那么答案就加上\(k \times w\)

但是还要注意我们把值域相同的缩成一个点,显然相同值的点之间连生成树边也是一种最优的情况

于是再对每一个出现的值\(w\),记录出现次数\(rnk[w]\),答案加上\(w \times (rnk[w]-1)\)

又跑进了rank1 hhh

代码

/*
  code by RyeCatcher
*/
const int maxn=100005;
const int inf=0x7fffffff;
int n,a[maxn],mx=-1;
int fa[maxn],rnk[maxn];
bool vis[maxn];
int get(int x){return (fa[x]==x)?fa[x]:(fa[x]=get(fa[x]));}
ll ans=0;
int main(){
    int x,y,p;
    //FO(gcd);
    read(n);
    for(ri i=1;i<=n;i++){
        read(a[i]);
        if(vis[a[i]]){
            rnk[a[i]]++;
        }
        else{
            vis[a[i]]=1;
            fa[a[i]]=a[i];
            rnk[a[i]]=1;
        }
        mx=max(mx,a[i]);
    }
    int pre=0;
    for(ri k=mx;k>=1;k--){
        pre=0,y=0;
        for(ri o=k;o<=mx;o+=k){
            if(vis[o]){
                if(!pre)pre=get(o);
                else {
                    x=get(o);
                    if(pre!=x){
                        y++;
                        if(rnk[pre]>rnk[x])fa[x]=pre;
                        else fa[pre]=x,pre=x;
                    }
                }       
            }
            
        }
        if(vis[k])ans+=1ll*(y+rnk[k]-1)*k;//别忘了乘以边权 
        else ans+=1ll*k*y;
    }
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/Rye-Catcher/p/9824106.html