Codeforces 1114B (贪心)

题面

传送门

分析

答案很好看出，显然是选最大的m*k个数

那么如何构造方案呢

我们把最大的m*k个数的位置标记为1，其他标记为0

从左到右维护一个ptr，记录有标记的数的个数，如果当前有m个有标记的数，则已经找到一个满足条件的区间分界点，break

这样的操作进行k-1次,由于ptr单调递增，时间复杂度为\(O(n)\)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
#define maxn 200005
using namespace std;
typedef long long ll;
int n,m,k;
struct node{
    int x;
    int id; 
}a[maxn];
int mark[maxn];
int cmp(node a,node b){
    return a.x>b.x;
}
vector<int>res;
map<int,int>cnt;
int main(){
    scanf("%d %d %d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].x);
        a[i].id=i;
    }
    sort(a+1,a+1+n,cmp);
    ll ans=0;
    for(int i=1;i<=m*k;i++){
        ans+=a[i].x;
        mark[a[i].id]=1;
    }
    int ptr=1;
    for(int i=1;i<k;i++){
        int len=0;
        while(ptr<=n){
            if(mark[ptr]){
                len++;
            }
            if(len==m) break;
            ptr++;
        }
        res.push_back(ptr);
        ptr++; 
    }
    printf("%lld\n",ans);
    for(int x : res){
        printf("%d ",x);
    }
}