2016 UESTC Training for Data Structures J - 郭大侠与Rabi-Ribi CDOJ 1334 优先队列

J -
郭大侠与Rabi-Ribi
就是有N只兔子，每只兔子会存在a[i]秒，价值为v[i]，然后每秒只能取一只兔子，问能取的兔子总价值的最大值是多少
用一个堆/优先队列维护就好了
首先我们先把兔子按存在时间升序排序，然后第一秒的时候，我们取第一只兔子，并放进堆中，堆是小根堆，按兔子的价值排序的，堆顶为价值最小的兔子
然后我们就一秒一秒的取，如果下一只兔子的存在时间>=当前时间，那就直接把下一只兔子放进来，然后如果下一只兔子的存在时间<当前时间，然后如果这只兔子比堆顶的兔子的价值大时，说明我们当时应该不取堆顶的兔子而取这一只兔子，于是就把堆顶的给pop，把这只兔子push就可以啦，同时维护一下总价值
然后把N只兔子遍历一遍，答案也就得到了
复杂度nlogn

代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define maxn 100005
struct Node
{
ll a, v;
bool operator <(Node x)const
{
return v > x.v;
}
}rabbit[maxn];
bool cmp(Node x, Node y)
{
return x.a < y.a;
}
int main()
{
//freopen("input.txt", "r", stdin);
//freopen("output.txt", "w", stdout);
int N;
scanf("%d", &N);
for (int i = 0; i < N; ++i)
scanf("%lld", &rabbit[i].a);
for (int i = 0; i < N; ++i)
scanf("%lld", &rabbit[i].v);
sort(rabbit, rabbit + N, cmp);
priority_queue<Node> Q;
ll time = 0, ans = 0;
Q.push(rabbit[0]);
++time, ans += rabbit[0].v;
for (int i = 1; i < N; ++i)
{
//printf("%lld\n", Q.top().v);
if (rabbit[i].a <= time&&rabbit[i].v>Q.top().v)
{
ans -= Q.top().v;
ans += rabbit[i].v;
Q.pop();
Q.push(rabbit[i]);
//++time;
}
else if (rabbit[i].a>time)
{
Q.push(rabbit[i]);
++time;
ans += rabbit[i].v;
}
}
printf("%lld\n", ans);
//system("pause");
//while (1);
return 0;
}