题目链接:https://codeforces.com/contest/1114/problem/B

       题意是输入n,m,k,然后输入n个数,让把这些数分成k组,每组至少m个数,对于每一组求出前m个最大的数的和,然后输出这些和的最大值,以及分组的位置。

       根据题意,最优解一定是这些数从大到小排序以后的前m*k个,然后就是给这些数分组,我们在计算前m*k个数的和的时候,做一个标记,然后遍历数组,如果一个区间里包含了m个有标记的数,就输出这个区间的最右端就好了。


AC代码:

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,k;
ll pre[200005], a[200005];

bool cmp(ll a, ll b){
	return a > b;
}

int main()
{
	scanf("%d%d%d",&n,&m,&k);
	vector<ll> v, G;
	ll sum = 0;
	for(int i=1;i<=n;i++){
		scanf("%lld",&pre[i]);
		a[i] = pre[i];
	}
	map<ll,int> ma;
	sort(pre + 1, pre + 1 + n, cmp);
	for(int i=1;i<=k*m;i++){
		sum += pre[i];
		ma[pre[i]] ++;
	}
	printf("%lld\n", sum);
	int pos = 0;
	int cnt = 0;
	for(int i=1;i<=n;i++){
		if(pos == k - 1)break;
		if(ma[a[i]] != 0){
			cnt ++;
			ma[a[i]] --;
		}
		if(cnt == m){
			printf("%d ",i );
			pos ++;
			cnt = 0;
		}
	}
	return 0;
}