CF 853A Planning

计划 题解

发现每架飞机等待单位时间都会产生一定费用，单位时间只能消除一架飞机，那么显然消除费用最大的飞机，用堆来维护这个最大值。当然，堆要用struct，因为要输出方案，必须记录编号，重载运算符不要弄反就是了。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=300005;
struct node{
    int tim,val;
    friend bool operator < (const node &a,const node &b) {
        return a.val<b.val;
    }
    node(int t,int v) : tim(t) ,val(v) {}
};
priority_queue<node>q;
int n,ans,res[N],k;
signed main() {
    scanf("%lld%lld",&n,&k);
    for(int i=1,x;i<=k;++i) {
        scanf("%d",&x);
        q.push(node(i,x));
    }
    for(int i=k+1,x;i<=n;++i) {
        scanf("%lld",&x);
        q.push(node(i,x));
        ans+=(i-q.top().tim)*q.top().val;
        res[q.top().tim]=i;
        q.pop();
    }
    for(int i=n+1;i<=n+k;++i) {
        ans+=(i-q.top().tim)*q.top().val;
        res[q.top().tim]=i;
        q.pop();
    }
    printf("%lld
",ans);
    for(int i=1;i<=n;++i)
        printf("%lld ",res[i]);
    return 0;
}