题目意思是给出n个数,每次挑两个数,满足加和小于等于k,然后求出这两个数的积。最多操作m次。
求多次的积的最大的和。
首先先明确,最大的数,乘以满足条件的最大的数必然得到的积最大。比如k=8,我有1,2,3,5。那么我必然选3,5.
思路是先模拟操作,求出每次的积保存起来再加和。因为,第一次的数求积并不一定是所有积里最大的一个。如k=8,数组是1,2,3,5,6.第二组3*5=15,第一组2*6=12.这就要求保存所有的积,并且排序求大值。
#include<bits/stdc++.h>
using namespace std;
multiset <long long> s;
multiset <long long> :: iterator pos1,pos2;
long long n,m,k;
long long ans[100005];
int main()
{
int t;
cin>>t;
while(t--)
{
long long cnt=0;
cin>>n>>m>>k;
s.clear();
for(int i=0;i<n;i++){
long long x;
cin>>x;
if(x<=k)
s.insert(x);//set自动排序
}
while(1){//先不断操作,让这个集合操作到不能再操作
if(s.size()<2)
break;
pos1=s.end();
pos1--;
long long x=*pos1;//先找出最大的,然后找能和这个数相加不超过k的最大的数
s.erase(pos1);//清除这个数
pos2=s.lower_bound(k-x);//找到第一个大于等于k-x的数
if(pos2==s.end())//所有元素都小于要求的数,就取最大的那个
pos2--;
if(pos2==s.begin()&&*pos2>(k-x))//判断,最小的数都不满足条件,直接continue
continue;
if(*pos2<=k-x){//判断完两个极端。现在的满足条件,放入结果
ans[cnt++]=x*(*pos2);
s.erase(pos2);
}
else {//不满足的必然是前推一个值
pos2--;
ans[cnt++]=x*(*pos2);
s.erase(pos2);
}
}
sort(ans,ans+cnt,greater<long long>());
long long result=0;
for(int i=0;i<min(m,cnt);i++)
result+=ans[i];
cout<<result<<endl;
}
return 0;
}