【UESTC 1334 郭大侠与Rabi-Ribi】+ 贪心 + 优先队列

郭大侠与Rabi-Ribi

Time Limit: 3000/1000MS (Java/Others) Memory Limit: 65535/65535KB (Java/Others)

Submit Status

title

最近郭大侠迷上了玩Rabi-Ribi这个游戏。

Rabi-Ribi呢，是一个打兔子的动作冒险游戏，萌萌哒的兔子在地上跑来跑去，好萌好萌呀~

这个游戏是这样玩的，郭大侠作为一个主角，拿着一个小锤子，他的目标是敲晕兔子，然后最后把这些敲晕的兔子都带回家。

当然咯，郭大侠想带回的兔子的总价值最高~

但是，兔子实在是太多了，郭大侠的锤子每一秒钟只能敲晕一只兔子，而且每一只兔子只会在地面上逗留a[i] a[i]秒，在a[i] a[i]秒之后，这一只兔子就会跑回自己的小窝里面。

所以郭大侠面临一些抉择，希望你能帮助他。

Input

第一行包含一个整数N N表示有N N个兔子在地上跑来跑去。

第二行N N个用空格分隔的整数a[i] a[i]表示第i只兔子冒出后停留的时间

第三行N N个用空格分隔的整数v[i] v[i]表示第i只兔子的价值。

1≤N≤100000 1≤N≤100000

1≤a[i]≤5000 1≤a[i]≤5000

1≤v[i]≤1000 1≤v[i]≤1000

Output

输出郭大侠最多能获得的价值是多少

Sample input and output

Sample Input

Sample Output

5
5 3 6 1 4
7 9 2 1 5 24
3
1 1 1
1 2 3 3

最后一秒开始，每次当前秒可以敲时就加入价值优先队列，每次取栈顶元素

AC代码：

#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long LL;
struct node{
    int vl,t;
}st[100010];
bool cmp(node i,node j){
    return i.t < j.t;
}
priority_queue<int> q;
int main()
{
    int N;
     scanf("%d",&N);
      for(int i = 0; i < N; i++)
         scanf("%d",&st[i].t);
      for(int i = 0; i < N; i++)
         scanf("%d",&st[i].vl);
      sort(st,st + N,cmp);
     int nl = st[N - 1].t,pl = N - 1;
        LL ans = 0;
    while(!q.empty()) q.pop();
      while(nl){
        while(pl >= 0 && st[pl].t >= nl) q.push(st[pl].vl),pl--;
        nl--;
        if(!q.empty())
        ans += q.top(),q.pop();
    }
    printf("%lld\n",ans);
    return 0;
}