最近复习期末考试,在忙乱之中挤出一点时间做题,突然发现自己好像只会套模板了(ㄒoㄒ)思维能力还是不足,比如下面这题:
拿到题目,想当然认为只要选取前m大的不就行了吗,最后WA……
这种想法并没有考虑到负负得正,比如
6 5
2 -2 4 2 -4 -2
这组数据,按照上面的想法会得到64的错误结果,实际答案为(-4)x(-2)x422=128。
那怎么办呢,只能暴力搜索了吗?
void dfs(int i,int st,int tot,int mul=1) {
if(i>=tot) {
res=max(mul,res);
return;
}
for(int j=st;j<n;j++) {
dfs(i+1,j+1,tot,mul*a[j]);
}
}
这题数据很小,暴力搜索还真能过,估计考场上遇到这题直接就上这种做法了。但数据量一大就肯定不行了,所以必须另想办法。
翻看题解后 发现一种思路:
分为两种情况:
第一种:m为偶数,取2k个正数和n-2k个负数,依次比较, 取最大值即可 第二种:m为正数,取2k+1个正数和n - (2k+1)个负数,依次比较,取最大值即可
分类讨论有点麻烦,下面是我由此想到的思路:以选取负数的个数为基准,考虑到负负得正,所以可以选偶数个负数,剩下的全选正数,并比较结果即可。当然还要考虑正数不够用的情况。代码如下:
int neg=0,pos=0,ans;
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++) {
scanf("%d",&a[i]);
(a[i]>=0)?pos++:neg++;
}
sort(a,a+n);
ans=a[n-1]; //防止出现选1个的情况
for(int k=0; 2*k<=neg; k++) { //2*k为负数个数
if(m-2*k<=pos) { //如果正数够用
int t1=negtive(2*k),t2=positive(m-2*k);
//分别为计算负数与正数乘积的函数
ans=max(ans,t1*t2);
}
}
printf("%d\n",ans);
下为两个函数的代码:
int negtive(int k) {
int res=1;
for(int i=0; i<k; i++)
res*=a[i];
return res;
}
int positive(int k) {
int res=1;
for(int i=n-1; n-1-i<k; i--)
res*=a[i];
return res;
}
更正:本方法不能解决全部为负数且要选奇数个的情况。这种情况需要特判。