乘积最大
给定N个整数A1, A2, … AN。请你从中选出K个数,使其乘积最大。
请你求出最大的乘积,由于乘积可能超出整型范围,你只需输出乘积除以1000000009的余数。
注意,如果X<0, 我们定义X除以1000000009的余数是负(-X)除以1000000009的余数。
即:0-((0-x) % 1000000009)
【输入格式】
第一行包含两个整数N和K。
以下N行每行一个整数Ai。
对于40%的数据,1 <= K <= N <= 100
对于60%的数据,1 <= K <= 1000
对于100%的数据,1 <= K <= N <= 100000 -100000 <= Ai <= 100000
【输出格式】
一个整数,表示答案。
【输入样例】
5 3
-100000
-10000
2
100000
10000
【输出样例】
999100009
再例如:
【输入样例】
5 3
-100000
-100000
-2
-100000
-100000
【输出样例】
-999999829
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 1000ms
思路:避免符号问题产生的麻烦,全程使用绝对值计算,负数另设一个标记变量进行标记。
struct node
{
int x;
int flag;
}a[N];
分类进行判断,有0的话标记一下,将n减去1,不存入数组。输入完毕后先判断此时n是否因为去掉0而小于k,若是,那么乘积一定为0。
for(int i=0;i<n;i++)
{
scanf("%d",&tem);
if(tem==0)
{
item=1;n--;i--;continue;
}
a[i].x=abs(tem);
if(tem<0)
a[i].flag=1;
else
a[i].flag=0;
}
if(n<k)
printf("0\n");
若不是再进行后续判断。按绝对值大小将数组排序。计算绝对值最大的k个数中负数的个数,若为奇数个,且负数个数小于n,那么此时有两种方案,一种是最大的k个数中找出绝对值最小的负数b1与前面绝对值最大的正数a1进行交换。另一种是从最大的k个数中找出最小的正数b2与前面绝对值最大的负数a2进行交换。然后分别判断它们是否可行,并找出最优方案
完整代码:
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
#define N 100005
#define mod 1000000009
struct node
{
int x;
int flag; //标记变量
}a[N];
int n,tem,item=0,k;
int cmp(node m,node n)
{
return m.x<n.x;
}
void solve()
{
sort(a,a+n,cmp);
int fushu=0,flag2=0;
ll ans=1;
for(int i=n-1;i>=n-k;i--)
{
if(a[i].flag)
fushu++;
}
//cout<<fushu<<endl;
if(fushu%2 && fushu<n) //若负数的个数为奇数个并且小于总个数
{
int a1=-1,a2=-1,b1=-1,b2=-1; //- + + -
for(int i=n-k;i<n;i++) //a1 b1 a2 b2
if(a[i].flag)
{
b2=i;break;
}
for(int i=n-k;i<n;i++)
if(!a[i].flag)
{
b1=i;break;
}
for(int i=n-k-1;i>=0;i--)
if(a[i].flag)
{
a1=i;break;
}
for(int i=n-k-1;i>=0;i--)
if(!a[i].flag)
{
a2=i;break;
}
//cout<<a1<<" "<<b1<<" "<<a2<<" "<<b2<<endl;
if((b2<0 || a2<0) && a1>=0 && b1>=0) //第一种方案可行
a[b1].x=a[a1].x;
else if((a1<0 || b1<0 ) && a2>=0 && b2>=0) //第二种方案可行
a[b2].x=a[a2].x;
else if(a1>=0 && a2>=0 && b1>=0 && b2>=0) //两种方案都可行,找出最优方案
{
if(a[a1].x*a[b2].x>a[b1].x*a[a2].x)
a[b1].x=a[a1].x;
else
a[b2].x=a[a2].x;
}
else //两种方案都不可行,最终结果只能为负,进行标记
flag2=1;
}
//cout<<flag2<<endl;
if(flag2 || n==fushu)
{
if(item) //能输出0就输出0
{
printf("0\n");return;
}
printf("-");
for(int i=0;i<k;i++) //找绝对值最小的k个数相乘得到最大负数
{
ans*=a[i].x;
ans%=mod;
}
cout<<ans<<endl;return;
}
for(int i=n-1;i>=n-k;i--)
{
ans*=a[i].x;
ans%=mod;
}
cout<<ans<<endl;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++)
{
scanf("%d",&tem);
if(tem==0)
{
item=1;n--;i--;continue;
}
a[i].x=abs(tem);
if(tem<0)
a[i].flag=1;
else
a[i].flag=0;
}
if(n<k)
printf("0\n");
else
{
solve();
}
return 0;
}