CodeForces - 853A Planning（贪心）

Planning

题目链接：Planning

题意：飞机场原定计划从第一分钟开始每分钟起飞一架飞机，但是由于某些原因导致前k分钟无法起飞飞机。给出每架飞机每延误一分钟的损失${cost_i}$，问所有飞机都起飞后的最小损失为多少

思路：对于k+1到k+n的每一时刻，每次把将要延误的所有飞机加入队列中，然后取出cost最大的，让其在这个时刻起飞，最后所得的答案就是最小的损失

代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;

typedef long long LL;
const int maxn=3e5+10;
struct node
{
    int id,cost;
    bool operator < (const node&A)const
    {
        return cost<A.cost;
    }
} p[maxn];

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; ++i)
        scanf("%d",&p[i].cost),p[i].id=i;
    priority_queue<node>q;
    for(int i=1; i<=k; ++i)
        q.push(p[i]);
    LL sum=0;
    for(int i=k+1; i<=n+k; ++i)
    {
        if(i<=n)
            q.push(p[i]);
        node now=q.top();
        q.pop();
        sum+=(LL)(i-now.id)*now.cost;
        p[now.id].id=i;
    }
    printf("%lld\n",sum);
    for(int i=1; i<=n; ++i)
        printf("%d ",p[i].id);
    return 0;
}