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

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