UPCOJ 5502(优先队列)

题目链接：http://exam.upc.edu.cn/problem.php?id=5502

 

题目大意：打地鼠，每只地鼠都从0时刻冒出来，但是停留一定的时间有一定的分值，问你怎么打分数最高

 

题目思路：设置一个now记录秒，对地鼠按照时间升序，如果时间相同分值降序来排序，然后从第一只地鼠到最后一只地鼠，如果now比地鼠停留的时间久，那么很开心进入队列，如果不是，那么就需要把队列中最小的那个数拿出来比一下，如果比那个数大，那队列里那个就弹出来，这个加进去，反之直接continue

 

以下是代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define MAXN 100005
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
struct node{
    int t,p;
}a[MAXN];
bool cmp(node x,node y){
    if(x.t==y.t)return x.p>y.p;
    return x.t<y.t;
}
priority_queue<int,vector<int>,greater<int> >q;
int main(){
    int n;
    while(~scanf("%d",&n)){
        rep(i,1,n){
            scanf("%d",&a[i].t);
        }
        rep(i,1,n){
            scanf("%d",&a[i].p);
        }
        sort(a+1,a+n+1,cmp);
        while(!q.empty())q.pop();
        int now=0,ans=0;
        rep(i,1,n){
            if(now<a[i].t){
                now++;
                q.push(a[i].p);
                ans+=a[i].p;
            }
            else{
                int temp=q.top();
                if(temp>a[i].p)continue;
                ans+=a[i].p-temp;
                q.pop();
                q.push(a[i].p);
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}