Codeforces 1395 D. Boboniu Chats with Du（贪心&&枚举答案）

传送门

题意：

思路：

先把大于m的放在b数组，小于m的放在a数组

枚举一下最终结果中用了几个b，取一个最大值即可
看代码应该能明白（有注释）

代码：

int a[MAXN],b[MAXN];
ll sum[MAXN];
ll sum2[MAXN];
int cmp(int a,int b){
	return a>b;
}
int main()
{
    std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    ll n,d,m;
    cin>>n>>d>>m;
    int x;
    int cnt=0,cc=0;
    for(int i=1;i<=n;i++){
    	cin>>x;
    	if(x>m)b[++cc]=x;
    	else a[++cnt]=x;
    }
    sort(a+1,a+cnt+1,cmp);//从大到小排序
    for(int i=1;i<=cnt;i++)sum[i]=sum[i-1]+a[i];
    sort(b+1,b+cc+1,cmp);
	for(int i=1;i<=cc;i++)sum2[i]=sum2[i-1]+b[i];
 
	ll ans=sum[cnt];
	for(int i=1;i<=cc;i++){//枚举最终用了几个可以禁言的
		ll p=1ll*(i-1)*d+i;
		if(p>n)continue;
		ll k=max(0ll,1ll*(i-1)*d-(cc-i));//k表示消耗的a的个数，cc-i是消耗的b数组里的，因为可能剩下的b比较多，不消耗a，跟0取个最大值
		ll tmp=sum2[i];//禁言部分的价值
		tmp+=sum[cnt-k];//a剩余部分的价值，k表示消耗的a的个数
		ans=max(ans,tmp);
	}
	cout<<ans<<endl;
    return 0;
}