传送门:http://codeforces.com/problemset/problem/853/A
这题首先是用贪心和并查集,虽然不知道怎么证明……但是贪心应该是对的,只要让花费大的飞机尽量早的起飞就可以了。主要难点是怎么判断这个时间点被占用了,我们可以用并查集来解决,ne数组中的ne[i]表示在 i 时间之后最近的可以起飞的时间点。
首先可以ne数组的每一个元素初始化为自身,表示现在这个时间点是可以起飞的。如果贪心的话,当然按时起飞是最优的啦,初始化为自身也有这样的考虑。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 3e5+10;
struct node
{
ll c;
int pos;
}node[MAXN];
int n,k;
int ans[MAXN];
int ne[MAXN*2]; // 有可能出现当前所有飞机全部不能按时起飞的情况,所以必须全部后延
bool cmp(struct node a,struct node b)
{
return a.c> b.c;
}
void init()
{
for(int i = 0;i<= MAXN*2;i++)
ne[i] = i;
return ;
}
int Find(int x)
{
return ne[x] == x? x : ne[x] = Find(ne[x]);
}
int main()
{
//freopen("input.txt", "r", stdin);
scanf("%d %d", &n, &k);
init();
for(int i = 1;i<= n;i++)
{
scanf("%I64d",&node[i].c);
node[i].pos = i;
}
ll sum = 0;
sort(node+1,node+n+1,cmp);
for(int i = 1;i<= n;i++)
{
int fa = Find(max(node[i].pos,k+1)); //检查当前时间点是否大于k+1,如果小于的话必须延时起飞
ans[node[i].pos] = fa; // ans为当前时间之后第一个可用的起飞时间
sum+= (fa-node[i].pos)*node[i].c;
ne[fa] = Find(fa+1); // 当fa时间点已经被占用后,ne[fa]更新为fa的下一个时间点的最近的时间(汗……)
}
cout<<sum<<endl;
for(int i = 1;i<= n;i++)
if(i == n)
printf("%d\n",ans[i]);
else
printf("%d ",ans[i]);
return 0;
}