打地鼠游戏

打地鼠游戏

时间限制: 1 Sec  内存限制: 128 MB

题目描述

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

输入

输入包含3行，第一行包含一个整数n（1<=n<=100000）表示有n个地鼠从地上冒出来，第二行n个用空格分隔的整数表示每个地鼠冒出后停留的时间（Maxt<=50000），第三行n个用空格分隔的整数表示每个地鼠被敲击后会增加的分值v（v<=1000）。每行中第i个数都表示第i个地鼠的信息。

输出

输出只有一行一个整数，表示所能获得的最大游戏总分值。

样例输入

5
5 3 6 1 4
7 9 2 1 5

样例输出

24

提示

30%的数据保证n<=100, t<=500,v<=50
60%的数据保证 n<=10000,t<=3000,v<=500
100%的数据保证 n<=100000,t<=5000,v<=1000

维护一个堆，把价值小的去除

#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;

const int MAXN = 1e5+5;
struct Rat{
	int t,v;
}rat[MAXN];

bool compare(Rat a,Rat b){
	return a.t<b.t;
}

int main(){
	int n,now=0,ans=0;
	priority_queue<int,vector<int>,greater<int> > heap;
	scanf("%d",&n);
	for(int i=0;i<n;i++) scanf("%d",&rat[i].t);
	for(int i=0;i<n;i++) scanf("%d",&rat[i].v);
	sort(rat,rat+n,compare);
	
	for(int i=0;i<n;i++){
		if(now<rat[i].t){
			now++;
			ans+=rat[i].v;
			heap.push(rat[i].v);
		}
		else{
			int t=heap.top();
			if(t<rat[i].v){
				ans=ans-t+rat[i].v;
				heap.pop();
				heap.push(rat[i].v); 
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}