标题:乘积最大
给定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
请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
注意:
main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中 #include <xxx>
不能通过工程设置而省略常用头文件。
提交程序时,注意选择所期望的语言类型和编译器类型。
思路:
感觉难点在分类讨论的时候要把所有情况都考虑全,首先把这些数按照绝对值从大到小排序:
①当n==k时,只能全选了
②如果n个数全是正数,选择前k个即可
③如果n个数全是负数,如果k是偶数,选择前k个即可;如果k是奇数,选择后k个即可
④剩下的情况就是有正有负了,先看前k个数,如果负数个数为偶数,选择前k个即可;如果负数个数为奇数,则比较前面最后一个负数与后面第一个负数的乘积和前面最后一个正数与后面第一个正数的乘积哪个大,取大的即可
代码:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define N 100000
#define MOD 1000000009
struct number
{
int sign;
long long num;
}a[N];
bool cmp(const number &x,const number &y)
{
return x.num>y.num;
}
int main()
{
int n,k,cnt=0;
long long t,ans=1;
cin>>n>>k;
for(int i=0;i<n;i++)
{
cin>>t;
if(t<0)
{
a[i].sign=-1;
a[i].num=-t;
cnt++;
}
else
{
a[i].sign=1;
a[i].num=t;
}
}
if(n==k)
{
for(int i=0;i<k;i++)
ans=(ans*a[i].num)%MOD;
if(cnt&1)
ans=-ans;
}
else
{
sort(a,a+n,cmp);
if(cnt==n)//全是负数
{
if(k&1)
{
for(int i=n-1;i>n-1-k;i--)
ans=(ans*a[i].num)%MOD;
ans=-ans;
}
else
{
for(int i=0;i<k;i++)
ans=(ans*a[i].num)%MOD;
}
}
else if(cnt==0)//全是正数
{
for(int i=0;i<k;i++)
ans=(ans*a[i].num)%MOD;
}
else
{
int nega=-1,posi=-1;
cnt=0;
for(int i=0;i<k;i++)
{
if(a[i].sign==-1)
{
//nega记录前K个中最后一个负数的下标
//cnt记录前k个中负数的个数,每满2个就乘一次
cnt++;
if(nega==-1)
nega=i;
else
{
if((cnt&1)==0)//保持负数是一对的样子乘上去
{
ans=(ans*a[nega].num)%MOD;
ans=(ans*a[i].num)%MOD;
}
nega=i;
}
}
else
{
//posi记录前k个中最后一个正数的下标
//每次只乘上前一个正数,留最后一个正数作交换
if(posi==-1)
posi=i;
else
{
ans=(ans*a[posi].num)%MOD;
posi=i;
}
}
}
if((cnt&1)==0)//负数为偶数个,则直接取前k个数即可,把最后一个正数乘上即可
ans=(ans*a[posi].num)%MOD;
else
{
int nega1=-1,posi1=-1;
for(int i=k;i<n;i++)
{
if(a[i].sign==-1)//找绝对值最大的负数
{
if(nega1==-1)
nega1=i;
}
else//找最大的正数
{
if(posi1==-1)
posi1=i;
}
if(nega1!=-1&&posi1!=-1)
break;
}
if(nega1!=-1&&posi1!=-1)
{
//比较是前k个中最后一个负数和后面绝对值最大的负数乘积大
//还是前k个中最后一个正数和后面最大的正数乘积大
if(a[nega].num*a[nega1].num>a[posi].num*a[posi1].num)
{
ans=(ans*a[nega].num)%MOD;
ans=(ans*a[nega1].num)%MOD;
}
else
{
ans=(ans*a[posi].num)%MOD;
ans=(ans*a[posi1].num)%MOD;
}
}
else if(posi1!=-1)//后面只有正数了
{
ans=(ans*a[posi].num)%MOD;
ans=(ans*a[posi1].num)%MOD;
}
else//后面只有负数了
{
ans=(ans*a[nega].num)%MOD;
ans=(ans*a[nega1].num)%MOD;
}
}
}
}
cout<<ans<<endl;
return 0;
}