题目链接
题意:
给你n个数,将他分成k组,求每一组的最大值和最小值相加得到的值的总和是多少。
思路:
首先想要总和最大,那么我们要尽可能使最大值最大,最小值尽可能大,我们先将两个数组排序,如果有k数组为1的情况,那么最大最小值相同,所以我们把最大的值放入1的堆中。当需要分的堆大于1时,首先我们要确保最大值为最大的,其次,最小值尽可能大,如果我们放入所有数字的最小值,那么其余的值不论有多小都不会对答案有影响,所以我们要利用这一点,放入一个最大值和最小值之后,把剩下的数从最小开始放入该数组中,使得下一个数组的最小值尽可能大。
代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=2e5+5;
const int mod=998244353;
const int inf=0x7fffffff;
const double pi=3.1415926535;
using namespace std;
int cmp(int a,int b)
{
return a>b;
}
signed main()
{
IOS;
int t;
cin>>t;
while(t--)
{
int n,k,num=0,arr[N],brr[N];
cin>>n>>k;
int l=0,r=n-1;
for(int i=0;i<n;i++)
{
cin>>arr[i];
}
sort(arr,arr+n);
for(int i=0;i<k;i++)
{
cin>>brr[i];
if(brr[i]==1)
{
num+=arr[r]*2;
r--;
}
}
sort(brr,brr+k,cmp);
for(int i=0;i<k;i++)
{
if(brr[i]==1)
{
break;
}
else
{
num+=arr[l]+arr[r];
r--;
l+=brr[i]-1;
}
}
cout<<num<<endl;
}
return 0;
}