【Codeforces】Round #538 (Div. 2) 1114B Yet Another Array Partitioning Task（数组+贪心）

原题链接：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, #