CodeForce 853A 解题报告 （贪心 并查集）

传送门：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':' ' );
    }
}