原题链接:B. Yet Another Array Partitioning Task
就不贴中文了,我也是直接看英文的
题意
这题题意很简单输入n,m,k
然后输入一组长度为n的序列,要求分成k组,每组的前m个最大值加起来的值要最大
题解
比赛的时候没想打结构体排序,只想着一次sort能不能完事,后来听了队友一顿操作,感受到了自己思路上的不足
#include<bits/stdc++.h>
using namespace std;
int n,m,k,a[200005],id[200005];
long long sum;
bool cmp(int x,int y) {
return a[x]>a[y];
}
int main() {
cin>>n>>m>>k;
int i;
for(i=1; i<=n; i++) {
cin>>a[i];
id[i]=i;
}
sort(id+1,id+n+1,cmp);
for(i=1; i<=m*k; i++)sum+=a[id[i]];
cout<<sum<<'\n';
sort(id+1,id+m*k+1);
for(i=m; i<m*k; i+=m)cout<<id[i]<<" ";
return 0;
}//By py_ultron, contest: Codeforces Round #538 (Div. 2), problem: (B) Yet Another Array Partitioning Task, Accepted, #