传送门:http://codeforces.com/problemset/problem/853/A
这题首先是用贪心和并查集,虽然不知道怎么证明……但是贪心应该是对的,只要让花费大的飞机尽量早的起飞就可以了。主要难点是怎么判断这个时间点被占用了,我们可以用并查集来解决,Next数组中的Next[i]表示在 i 时间之后最近的可以起飞的时间点。相当于如果 i 时刻已经被占用的话,那么Next[ i ]的值就不再是 i ,而是下一个最近的可起飞时刻。这个题体现了并查集,作为一种树形的结构,其实也可以是有方向性的。
首先可以Next数组的每一个元素初始化为自身,表示现在这个时间点是可以起飞的。如果贪心的话,当然按时起飞是最优的啦,初始化为自身也有这样的考虑。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN = 300010;
typedef long long LL;
int Next[MAXN * 2], TimeTable[MAXN];
struct Plane{
LL cost, time; // 这里要注意一发,这题输入数据很大,最好全部调成LL
}plane[MAXN];
int Find(int x){
return Next[x] == x ? x :Next[x] = Find(Next[x]);
}
void Init(int n, int k){
for(int i = 1;i <= n+k;++i)
Next[i] = i;
}
bool cmp(Plane a, Plane b){
return a.cost > b.cost;
}
int main(){
//freopen("input.txt", "r", stdin);
int n, k;
LL ans = 0;
scanf("%d %d", &n, &k);
Init(n, k);
for(int i = 1;i <= n;++i){
scanf("%I64d", &plane[i].cost);
plane[i].time = i;
}
sort(plane + 1, plane + 1 + n, cmp);
for(int i = 1;i <= n;++i){
int temp;
if( plane[i].time <= k )
temp = Find(k+1); // 这里要注意一发,一定是K + 1,K时刻是不能起飞的。
else
temp = Find(plane[i].time);
ans += (plane[i].cost * (temp - plane[i].time) * 1LL);
TimeTable[plane[i].time] = temp;
Next[temp] = Find(temp + 1); //这里意思是temp时刻已被占用,所以temp时刻下一可用时刻变为了Find(temp + 1)
}
printf("%I64d\n",ans);
for(int i = 1;i <= n;++i){
printf("%d%c", TimeTable[i], i==n?'\n':' ' );
}
}