[蓝桥杯][算法提高]ADV-15 最大乘积 贪心 C++
资源限制
时间限制:1.0s 内存限制:512.0MB
问题描述
对于n个数,从中取出m个数,如何取使得这m个数的乘积最大呢?
输入格式
第一行一个数表示数据组数
每组输入数据共2行:
第1行给出总共的数字的个数n和要取的数的个数m,1<=n<=m<=15,
第2行依次给出这n个数,其中每个数字的范围满足:a[i]的绝对值小于等于4。
输出格式
每组数据输出1行,为最大的乘积。
样例输入
1
5 5
1 2 3 4 2
样例输出
48
题解:
因为有可能有负数,所以不能一直选最大的。
负数要想选择,必须两个两个的选择。正数只需要一个一个选择就好(这样才能保证每次选择的数是最优的)。
那么,先把数列从小到大排序,每次选两个最大数相乘,
然后比较前两个数的乘积和最后两个数的乘积,
如果前面两个数的乘积较大,则res*前面的两个数的乘积(两个负数相乘的结果)
但如果后面的两个数较大,则只需选择一个数与res相乘
选到m剩1时,不能选负数了,则需在右边选(较大)的数相乘
注意点:
n<=m:总共的数字的个数n≤要取的数的个数m,这说明取的个数必须大于或等于总的个数
C++示例代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int main(){
int T;
cin>>T;
int n,m;int num;
for(int t=0;t<T;t++){
cin>>n>>m;
vector<int>number;
for(int i=0;i<n;i++){
cin>>num;
number.push_back(num);
}
sort(number.begin(),number.end());
int left=0,right=number.size()-1;
int res=1;
while(m>0){
if(m>=2){
int t1 = number[left]*number[left+1];
int t2 = number[right]*number[right-1];
if(t1>t2){
res*=t1;
left+=2;
m-=2;
}else{
res*=number[right];
right--;
m--;
}
}else{
res*=number[right];
right--;
m--;
}
}
cout<<res<<endl;
}
return 0;
}