1369C - RationalLee (1400)

1369C - RationalLee

题意：
有N个物品，每个物品价值A[i] ，将这些物品分配给K个人，每个人要W[i]个物品，求每个人得到的物品价值的最大值和最小值之和的最大值

思路：
贪心思想，先让只要一个物品的人拿价值最大的东西，接着让剩下的人先从价值最大的物品开始每人取走一个，然后再让W小的人从剩下的物品中从最大价值的开始取

代码附：

#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e5+10;
int a[N],w[N];
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t,n,k;
    cin>>t;
    while(t--)
    {
        int ans=0,only=0;
        cin>>n>>k;
        for(int i=0; i<n; ++i)
            cin>>a[i];
        for(int i=0; i<k; ++i)
        {
            cin>>w[i];
            w[i]--;
            if(w[i]==0)
            {
                only++;
                k--;
                i--;
            }
        }
        sort(w,w+k);
        sort(a,a+n);
        while(only--)
        {
            ans+=2*a[n-1];
            n--;
        }
        for(int i=0;i<k;++i)
        {
            ans+=a[n-1];
            n--;
        }
        for(int i=0;n>0;++i)
        {
            ans+=a[n-w[i]];
            n-=w[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}