绝妙的手法(牛客小白月赛98)

题意:给定一个数组a和一个正整数m,操作:选择不同下标i,j,使得ai+=aj,问最少操作次数,得数组刚好有m个正数。

知识点:贪心 数学

分析:

记录cnt为数组的正整数个数

1.m>n 无解

2.cnt>m,将一些正数变成负数

3.当cnt==m时,输出0

4.当cnt<m时,将一些非正数变成正数

如果用贪心无法达到最大值。

eg.-3 1 1 将负数加上最大值的话,答案为4.

但是可以先将1+1=2 变成-3 2 1,然后再让-3加上两次2,一共花费3次操作。

最优解:用最大俩数相互叠加,叠加过程是斐波那契函数,遍历最大的数,算每个数到最大数需要的操作次数相加,每次遍历取最小操作次数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
ll a[N],n,m;
void fz(){
    sort(a+1,a+n+1);
    ll an1=a[n],an2=a[n-1];
    if(an1<=0){//全部都是负数或0,无论怎么加都不会变成正数
        cout<<"-1"<<endl;
        return;
    }
    ll t=0;ll ans=0;
    for(int i=n-m+1;i<=n;i++){
        if(a[i]<=0){
            ans+=abs(a[i])/an1+1;
        }
    }
    if(an2<=0){//如果最大两个数一正一负,先把负数变成正数
        t+=abs(an2)/an1+1;
        an2+=t*an1;
        a[n-1]=an2;
    }
    
    for(int j=1;;j++){
        ll fibo=an1;
        an1=an1+an2;
        an2=fibo;
        t++;
        ll tpp=t;
        for(int i=n-m+1;i<=n;i++){
            if(a[i]<=0){
                tpp+=abs(a[i])/an1+1;
            }
        }
        ans=min(ans,tpp);
        if(an1>2*abs(a[n-m+1]))break;//最小的负数只需一步即可变成就可以结束了
    }
    cout<<ans<<endl;
    return;
}
void zf(){
    sort(a+1,a+n+1);
    ll an1=a[1],an2=a[2];
    if(an1>=0){//全部都是正数或0,无论怎么加都不会变成负数
        cout<<"-1"<<endl;
        return;
    }
    ll t=0;ll ans=0;
    for(int i=1;i<=n-m;i++){
        if(a[i]>0){
            ans+=(abs(a[i])+abs(an1)-1)/abs(an1);
        }
    }
    if(an2>=0){//如果最小两个数一正一负,先把正数变成负数
        t+=(abs(an2)+abs(an1)-1)/abs(an1);
        an2+=t*an1;
        if(an2==0){
            an2+=an1;
            t++;
        }
        a[2]=an2;
    }
    
    for(int j=1;;j++){
        ll fibo=an1;
        an1=an1+an2;
        an2=fibo;
        t++;
        ll tpp=t;
        for(int i=1;i<=n-m;i++){
            if(a[i]>0){
                tpp+=(abs(a[i])+abs(an1)-1)/abs(an1);
            }
        }
        ans=min(ans,tpp);
        if(abs(an1)>2*abs(a[n-m]))break;//最小的负数只需一步即可变成就可以结束了
    }
    cout<<ans<<endl;
    return;
}
void sol(){
    cin>>n>>m;
    int cnt=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]>0)cnt++;
    }
    if(m>n)cout<<"-1"<<endl;
    else if(cnt==m)cout<<"0"<<endl;
    else if(cnt<m)fz();//负数变正数
    else if(cnt>m){//正数变成负数
        zf();
    }
}
int main(){
    int t;cin>>t;
    while(t--)sol();
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值