做这个题的过程非常逗逼,一开始胡乱想了个算法,结果交上去wa了,于是在discuss里面找了一堆数据,然后每个没过的数据,我就补一种情况。最后大概是所有常见的情况都补完了,那些数据也都过了,最后就AC了。。
总之代码写的非常乱,大体意思就是每选两个数比较大小,特判了需要0以及结果为负的情况。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <map>
#include <algorithm>
#define ll long long
#define INF 2139062143
#define MAXN 10005
// ios::sync_with_stdio(false);
using namespace std;
int a[MAXN],b[MAXN],c[MAXN];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int N,K;
// freopen("mtouoooy.txt","w",stdout);
while(scanf("%d%d",&N,&K))
{
if(!N&&!K) break;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
int na=0,nb=0,zero=0;
for(int i=1; i<=N; ++i)
{
int x;
scanf("%d",&x);
if(x>=0) a[++na]=x;
else b[++nb]=x;
if(x==0) zero++;
c[i]=x;
}
int sum=0;
if(N-zero<K)
{
sort(c+1,c+N+1,cmp);
for(int i=1; i<=K; ++i) sum+=c[i];
printf("%d\n",sum);
continue;
}
sort(a+1,a+na+1,cmp);
sort(b+1,b+nb+1);
int sa=1,sb=1;
bool ok=false;
bool oo=false;
int k=K;
if(K%2)
{
K--;
if(sa<=na)
{
if(!a[sa]) oo=true;
sum+=a[sa];
sa=sa+1;
}
else
{
sort(b+1,b+nb+1,cmp);
sum+=b[sb];
sb=sb+1;
ok=true;
}
}
while(K>0)
{
K-=2;
if(sa+1<=na&&sb+1<=nb)
{
if(a[sa]*a[sa+1]>=b[sb]*b[sb+1])
{
if(!a[sa]||!a[sa+1]) oo=true;
sum+=a[sa]+a[sa+1];
sa=sa+2;
}
else
{
sum+=b[sb]+b[sb+1];
sb=sb+2;
}
}
else if(sa+1<=na)
{
if(!a[sa]||!a[sa+1]) oo=true;
sum+=a[sa]+a[sa+1];
sa=sa+2;
}
else if(sb+1<=nb)
{
sum+=b[sb]+b[sb+1];
sb=sb+2;
}
else
{
if(!a[sa]) oo=true;
sum+=a[sa];
sa=sa+1;
sum+=b[sb];
sb=sb+1;
ok=!ok;
}
}
if((ok&&zero)||oo)
{
sum=0;
sort(c+1,c+N+1,cmp);
for(int i=1; i<=k; ++i) sum+=c[i];
}
printf("%d\n",sum);
}
return 0;
}