n个数中取k个使得乘积最大(c++实现)

题目:

给定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;
}

    原文作者:大整数乘法问题
    原文地址: https://blog.csdn.net/qq_31344665/article/details/79810353
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞