Codeforces1513 D. GCD and MST(思维,MST)

题意:

在这里插入图片描述

解法:
如果gcd=min,那么区间内的所有数一定都是min的倍数.

如果不考虑gcd=min的情况,一定是n-1条代价为p的边,
现在我们要做的就是用gcd=min的边替换代价为p的边,
用mark[i]表示边(i,i+1)是否被替换.

将序列变为二元组(a[pos],pos),按照a[pos]从小到大排序,
枚举pos,计算以pos为最小值时,能向左向右扩展的最大未标记区间[l,r],
这时[l,r]中包含pos的所有子区间都满足gcd=min,
那么[l,pos-1],[pos+1,r]的所有点都向pos连边即可,
每条边的代价为a[pos],
每条边会替换掉原来的一条p边,因此记得标记对应的mark[].

总复杂度O(n*log),复杂度瓶颈在排序上.
code:
#include<bits/stdc++.h>
#define int long long
#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
int mark[maxm];
int a[maxm];
PI e[maxm];
int n,p;
void solve(){
    cin>>n>>p;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        e[i]={a[i],i};
    }
    sort(e+1,e+1+n);
    int ans=0,need=n-1;
    for(int i=1;i<=n;i++)mark[i]=0;
    for(int i=1;i<=n;i++){
        int pos=e[i].second;
        if(mark[pos])continue;
        if(a[pos]>=p)break;
        for(int j=pos+1;j<=n;j++){
            if(mark[j-1]||a[j]%a[pos]!=0)break;
            mark[j-1]=1;
            ans+=a[pos];
            need--;
        }
        for(int j=pos-1;j>=1;j--){
            if(mark[j]||a[j]%a[pos]!=0)break;
            mark[j]=1;
            ans+=a[pos];
            need--;
        }
    }
    ans+=need*p;
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    int T;cin>>T;
    while(T--){
        solve();
    }
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值