Codeforces Round #538 (Div. 2) B 贪心 整体考虑

题目

题解思路

整体考虑整个最值，我们可以知道前KM大的数一定可以加上。区间分成k份，每份取前M个，整合一下就是 全部前KM大的数 。我们只需标记这些数，对每个区间都取到M个即可。

学长这里用了个骚操作来解决负数问题
这里B数组存的是编号，这样就是存储了A数组降序排的时候的原编号。
这里太妙了。

bool cmp( int t1 , int t2 )
{
    return a[t1] > a[t2];
}
    for (int i = 1 ; i <= n ; i++ )
    {
        cin>>a[i];
        b[i] = i;
        vis[i] = 0 ;
    }
    sort(b+1,b+1+n,cmp);

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

const  int  INF =  0x3f3f3f3f;
const int P = 1e9 + 7;
int a[200100];
int b[200100];
bool vis[200100];

bool cmp( int t1 , int t2 )
{
    return a[t1] > a[t2];
}

int main ()
{
    ios::sync_with_stdio(false);
    int n,m,k;
    long long sum = 0 ;
    cin>>n>>m>>k;
    for (int i = 1 ; i <= n ; i++ )
    {
        cin>>a[i];
        b[i] = i;
        vis[i] = 0 ;
    }
    sort(b+1,b+1+n,cmp);
    for (int i = 1 ; i <= k*m ; i++ )
    {
        sum += a[b[i]];
        vis[b[i]] = 1;
    }
    cout<<sum<<"\n";
    int t2 = 0 ;
    for ( int i = 1 ; i <= n ; i++ )
    {
        t2++;
        int t1 = 0;
        while(t1 != m && i <= n)
        {
            t1 += vis[i];
            i++;
        }
        i--;
        if ( t2 < k )
            cout<<i<<" ";
    }
    cout<<"\n";
    return 0 ;
}