任务 - 数字的最大 K 积
时间限制:1
内存限制:64 M
给定整数序列 N (1 ≤ N ≤ 10 May, | A i | ≤ 2.10 9) 和 K 的数量 (1 ≤ K ≤ N)。找出积最大的 K 个序列号。
输入数据:
第一行包含两个整数 N 和 K。
第二行列出了序列 A 的 N 个元素。
输出数据:
导出最大乘积。所以答案可能相当大,以 10^9+7 为模输出。
示例
输入数据的结果
3 2
-2 -3 3
答案 - 6
下面是我的尝试。这是一个错误,这是我不知道的。你能帮我找出我的决定中的错误吗?
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <vector>
using namespace std;
typedef vector<int> v1;
const int mod = 1000000007;
int n, k, pos1 = 0, pos2 = 0, negative = 0;
long long res = 1;
void QuickSort(v1 &a, int l, int r) {
int i = l, j = r, pivot = abs(a[l + ((r - l) >> 1)]);
do {
while (abs(a[i]) < pivot) i++;
while (abs(a[j]) > pivot) j--;
if (i <= j) {
int temp = a[i];
a[i++] = a[j];
a[j--] = temp;
}
} while (i < j);
if (l < j) QuickSort(a, l, j);
if (i < r) QuickSort(a, i, r);
}
long long product(v1 &a, v1 &b, int q, int j, char flag) {
long long res = 1; int temppos;
if (flag == 0 && j) {
temppos = b[pos1];
b[pos1] = j;
}
if (flag == 1 && q) {
temppos = b[pos2];
b[pos2] = q;
}
if (!pos2 && (k & 1)) {
for (int i = 1; i <= k; i++)
res = (long long)((res % mod)*1ll*(a[i] % mod)) % mod;
} else {
for (int i = 1; b[i] != 0; i++)
res = (long long)((res % mod)*1ll*(a[b[i]] % mod)) % mod;
}
if (flag == 0 && j) b[pos1] = temppos;
if (flag == 1 && q) b[pos2] = temppos;
return res;
}
int main()
{
v1 a(100002, 0);
v1 b(100002, 0); //index multiplied to the elements
cin >> n >> k;
for (int i = 1; i <= n; i++) cin >> a[i];
QuickSort(a, 1, n);
for (int i = n, j = 1; i > n - k; i--) {
b[j] = i;
if (a[i] < 0) {
pos1 = j; //index last positive number
negative++; //increase the counter negative numbers
}
else pos2 = j; //index last positive number
j++;
}
int j = n - k, q = j;
if (negative & 1) { //If an odd number of negative numbers
while (j > 0 && a[j] < 0) j--;
while (q > 0 && a[q] > 0) q--;
res = max(product(a, b, q, j, 0), product(a, b, q, j, 1));
} else res = product(a, b, q, j, 3);
cout << res << endl;
cin >> res;
return 0;
}