「NOIP2018模拟赛」 打地鼠游戏 - 贪心+并查集

题目描述

伟大的学长特别喜欢打地鼠游戏，这个游戏开始后，会在地板上冒出一些地鼠来，你可以用榔头去敲击这些地鼠，每个地鼠被敲击后，将会增加相应的游戏分值。可是，所有地鼠只会在地上出现一段时间（而且消失后再也不会出现），每个地鼠都在0时刻冒出，但停留的时间可能是不同的，而且每个地鼠被敲击后增加的游戏分值也可能是不同。

最近学长经常玩这个游戏，以至于敲击每个地鼠只要1秒。他在想如何敲击能使总分最大。

分析

很明显的一个贪心，尽量选择较后的时间打下去。先按照增加分值从大到小排序，再在停留时间上建立并查集，从1~n枚举，若f[getf(a[i].lim)]==0则之前没有时间可以让其打下去，否则就让f[getf(a[i].lim)]=getf(a[i].lim)-1，向前一个时间点合并，优化贪心。

此题还可用堆来优化贪心。

代码

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#define ll long long
using namespace std;
struct Node {
	int lim,sor;
}a[1000005];
int f[700005],n,ans;//注意是以时间为并查集的关键字 
int getf(int x) {
	return f[x]==x?x:f[x]=getf(f[x]);
}
bool cmp(Node a,Node b) {
	return a.sor>b.sor;//按照得分从大到小排序 
}
int main() {
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i].lim);
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i].sor);
	sort(a+1,a+n+1,cmp);
	for (int i=1;i<=700000;i++) f[i]=i;//并查集初始化
	for (int i=1;i<=n;i++) {
		int x=getf(a[i].lim);
		if (x!=0) {
			ans+=a[i].sor;
			f[x]=getf(x-1);
		}
	}
	printf("%d",ans);
	return 0;
}