Codeforces 854C Planning（贪心+堆）

　　贪心：让代价大的尽量移到靠前的位置。

　　做法：先让前k个数加进堆里，枚举k+1~n+k，每次把新元素加进堆后找到最大代价放在当前位置即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=;
struct poi{int c,pos;};
priority_queue<poi>q;
bool operator<(poi a,poi b){return a.c<b.c;}
int n,k;
int a[maxn],ansi[maxn];
ll ans;
void read(int &k)
{
    int f=;k=;char c=getchar();
    while(c<''||c>'')c=='-'&&(f=-),c=getchar();
    while(c<=''&&c>='')k=k*+c-'',c=getchar();
    k*=f;
}
int main()
{
    read(n);read(k);
    for(int i=;i<=n;i++)read(a[i]);
    for(int i=;i<=k;i++)q.push((poi){a[i],i});
    for(int i=k+;i<=n+k;i++)
    {
        if(i<=n)q.push((poi){a[i],i});
        poi t=q.top();q.pop();
        ans+=1ll*t.c*(i-t.pos);
        ansi[t.pos]=i;
    }
    printf("%I64d\n",ans);
    for(int i=;i<=n;i++)printf("%d ",ansi[i]);
}