Codeforces Round #652 (Div. 2)--C. RationalLee（贪心）

题目传送

题意：

有n个数字被分给k个人，每个人都有一个wi值（1<=i<=k）,表示获得数字的个数，每个人都有一个快乐值，表示获得数字的最大值和最小值的和，问k个人获得的快乐值的总和最大是多少。

思路：

当wi等于1的时候，第i个人获得的整数的最大值和最小值都是一样的，由此可知，当wi==1是，获得的整数越大，贡献值就越多。当wi越小，获得的整数越大，那么，总的的获得的整数的最大值之和，是最大的。然后要考虑最小值的情况，每间隔wi（1<=i<=k）个，找到最小值，然后相加，就能是总和最大。中间的整数没有用到，就可以忽略不计。

上代码：

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int t,n,m,b[N],ans,a[N];

bool cmp(int x,int y)
{
	return x>y;
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
         scanf("%d%d",&n,&m);
         memset(a,0,sizeof a);
         memset(b,0,sizeof b);
         ll res=0;
        
         for(int i=1;i<=n;i++)
         {
         	scanf("%d",&a[i]);
		 }
		 sort(a+1,a+1+n,cmp);// 从大到小 
		 for(int i=1;i<=m;i++)
		 {
		 	scanf("%d",&b[i]);
		 }
		 sort(b+1,b+1+m);// 从小到大
		 int last=0; 
		 for(int i=1;i<=m;i++)
		 {
		 	if(b[i]==1)last=i;//统计一下最后wi==1的位置
		 	res+=a[i];//前面k个数之和就是k个人获得的最大值之和
		 }
		 //printf("%lld\n",res);
		 
		 for(int i=m,j=n;i>=1;i--)
		 {
		 	if(b[i]==1){
		 	//因为wi==1所以它的最大值和最小值一样
		 		res+=a[last];last--;continue;
			 }
			 res+=a[j];
			 j-=b[i]-1;//因为k个人的最大值已经找到，最小值肯定在剩余的wi-1中寻找最小值，所以是每间隔wi-1个取得的数，能使和最大
			 printf("%d %d\n",j,a[j]);
		 }
		 printf("%lld\n",res);
	}
	return 0;
}