Codeforces 853 A. Planning
传送门:https://codeforces.com/problemset/problem/853/A
题目大意:
这个飞机场上有n架飞机,按计划从第一天开始,每天飞出一架,但飞机场比较任性,它前面k天不想让飞机走。但它不让它们走又得赔违约金,每架飞机每迟一天要ai的钱,于是,架不住肉疼,它想安排一下飞行顺序,使得每天依然只让一架走,但最后亏的钱最少。
主要想法:
这个肯定要用到排序,而且每次肯定用最大的数,这个让我不禁想起了优先队列。好的,大的方向有了,我们还得注意,飞机没事不会提前飞,在这个前提下,我们只要让违约金贵的尽量少拖延,于是,我们就计算在第k+i天哪些飞机可以飞,然后选出最贵的就可以了。
代码实现:
#include <bits/stdc++.h>
using namespace std;
struct node
{
int x,y;
bool operator < (const node & a)const
{
return x<a.x;
}
}b;
int main()
{
int n,k;
long long ans=(long long)0;
cin>>n>>k;
priority_queue<node>a;
vector<int>d(n);
for(int i=0;i<k;++i)
{
int c;
scanf("%d",&c);
b.x=c;
b.y=i;
a.push(b);
}
for(int i=k;i<n;++i)
{
int c;
scanf("%d",&c);
b.x=c;
b.y=i;
a.push(b);
ans+=(long long)a.top().x*(i-a.top().y);
d[a.top().y]=i+1;
a.pop();
}
for(int i=n;i<n+k;++i)
{
ans+=(long long)a.top().x*(i-a.top().y);
d[a.top().y]=i+1;
a.pop();
}
cout<<ans<<endl;
for(int i=0;i<n;i++)printf("%d ",d[i]);
return 0;
}