Codeforces Round #652 (Div. 2)-C. RationalLee

题目链接

题意：

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