传送门1:http://www.usaco.org/index.php?page=viewproblem2&cpid=118
传送门2:http://www.lydsy.com/JudgeOnline/problem.php?id=2590
又挂了一道贪心,好烦啊。
这一题应该要想到“撤回”操作就好办了。假设现在已经没有优惠券了,那么如果你想以优惠价格买下一头牛,就要撤回以前的用优惠券买的牛,具体就是加回p[i] - c[i]就行了,可以理解为花这么多钱买一张优惠券。然后就是维护3个小根堆了。
#include <cstdio> #include <queue> const int maxn = 50005; int n, k, ans; long long m, pp[maxn], cc[maxn]; bool book[maxn]; struct st1 { long long data; int id; bool operator<(const st1 & rhs) const { return data > rhs.data; } }; std::priority_queue<st1> p, c; struct st2 { long long data; bool operator<(const st2 & rhs) const { return data > rhs.data; } }; std::priority_queue<st2> coupon; int main(void) { freopen("coupons.in", "r", stdin); freopen("coupons.out", "w", stdout); scanf("%d%d%I64d", &n, &k, &m); while (k--) { coupon.push((st2){0}); } for (int i = 1; i <= n; ++i) { scanf("%I64d%I64d", pp + i, cc + i); p.push((st1){pp[i], i}); c.push((st1){cc[i], i}); } while (m > 0 && ans < n) { while (book[p.top().id]) { p.pop(); } while (book[c.top().id]) { c.pop(); } if (p.top().data < c.top().data + coupon.top().data) { m -= p.top().data; if (m <= 0) { break; } book[p.top().id] = true; p.pop(); } else { m -= (c.top().data + coupon.top().data); if (m <= 0) { break; } coupon.pop(); coupon.push((st2){pp[c.top().id] - c.top().data}); book[c.top().id] = true; c.pop(); } ++ans; } printf("%d ", ans); return 0; }