题意:给定一个数组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;
}