题目:
给定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
思路:
使用的贪心的思想。先对数组按绝对值大小升序排序,然后看右边k个数。如果这k个数的乘积是正数,当然就是我们要的结果。但如果是负数,就要想办法从前面那些数里找一个绝对值最大的正数来替换k个数中绝对值最小的负数。
当然了还有很多复杂的情况,比如前面那些数没有正数,就只能找一个绝对值最大的负数去替换后面k个数中的最小正数,但也有可能后面没有正数,这时说明,整个数组都没有正数,所有结果应该是绝对值最小的k个负数相乘(即此时最左边的k个数)。最复杂的情况是前面的数有正有负,后面k个数也有正有负,这时需要比较,从前面找正数来替换乘积最大还是找负数来替换乘积最大。
代码:
我们需要记录的是前面的数里绝对值最大的正数、负数和后k个数里最小的正数、负数(可能不存在)。
#include <iostream>
#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
int n, k;
long long ans = 1;
int l_neg = -1, l_pos = -1, r_neg = -1, r_pos = -1;
const int M = 1000000009;
bool cmp(int a, int b)
{
return abs(a) < abs(b);
}
int calc(vector<int> a)
{
long long temp = 1;
for(int i = n - k; i < n; i++)
temp = (temp * a[i]) % M;
return temp;
}
bool v_cmp(vector<int> a, vector<int> b)
{
int sum1 = 0, sum2 = 0;
for(int i = n - k; i < n; i++){
sum1 += abs(a[i]);
sum2 += abs(b[i]);
}
return sum1 > sum2;
}
int main() {
cin >> n >> k;
vector<int> a(n, 0);
for(int i = 0; i < n; i++)
cin >> a[i];
sort(a.begin(), a.end(), cmp);
for(int i = n - k; i < n; i++){
ans = (ans * a[i]) % M;
if(r_pos == -1 && a[i] > 0) r_pos = i;
if(r_neg == -1 && a[i] < 0) r_neg = i;
}
for(int i = n - k - 1; i >= 0; i--){
if(l_pos == -1 && a[i] > 0) l_pos = i;
if(l_neg == -1 && a[i] < 0) l_neg = i;
}
if(ans < 0 && n != k){
if(r_pos == -1){
if(l_pos == -1){
ans = 1;
for(int i = 0; i < k; i++)
ans = (ans * a[i]) % M;
}
else {
swap(a[l_pos], a[r_neg]);
ans = calc(a);
}
}
else {
if(l_neg == -1){
swap(a[l_pos], a[r_neg]);
ans = calc(a);
}
else if(l_pos == -1){
swap(a[l_neg], a[r_pos]);
ans = calc(a);
}
else {
vector<int> c1 = a;
vector<int> c2 = a;
swap(c1[l_neg], c1[r_pos]);
swap(c2[l_pos], c2[r_neg]);
ans = v_cmp(c1, c2) == true? calc(c1): calc(c2);
}
}
}
cout << ans << endl;
return 0;
}