Codeforces C. RationalLee (思维) (Round #652 Div.2)

传送门

题意： 给你n个数，有m个朋友，每个朋友要拿其中w[i]个数，每个人获得的幸福值是他拿的数的最大值与最小值的和，如果只有一个数最大值最小值都是它，让你求出所有人能获得的最大幸福值是多少？

思路：

• 由于每个朋友的幸福值是其得到的max和min的和，那么就要保证其最大值经量大
• 将a[]和w[]都先按升序排列，将依次最大的a[i]分给w值为1的朋友，这样他的幸福值就是max*2，贡献值增长最快。
• 将所有w[i]为1的处理完后，再将a数组的max和min分配给w[i]最大的朋友，剩余的w[i] - 2个数就分配最小的w[i] - 2个数(因为中间的数的贡献为0)，依次下去所求就是答案。

代码实现：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int t, n, k;
int a[N], w[N];

int cmp(iint a, int b)
{
    return a > b;
}

signed main()
{
	IOS;
	cin >> t;
	while(t --){
		cin >> n >> k;
		for(int i = 0; i < n; i ++)
            cin >> a[i];
		for(int i = 0; i < k; i ++)
		    cin >> w[i];
        
		sort(a, a + n); sort(w, w + k, cmp);
		int ans = 0, p = n - 1;
		//将前k大的数分配给每个人一个
		for(int i = 0, j = n - 1; i < k; i ++, j --)
			ans += a[j];
        
		for(int i = 0, j = 0; i < k; i ++){	
			if(w[i] == 1){  //w[i]==1时，要特殊处理
                ans += a[p]; 
                p --;
                continue;
            }
			ans += a[j];	//加上w[i]的最小值
			j += w[i] - 1;	//因为中间的数没有贡献，可直接略去
		}
		cout << ans << endl;
	}
	return 0;
}