洛谷P3045 [USACO12FEB]牛券Cow Coupons

题目：洛谷P3045 [USACO12FEB]牛券Cow Coupons

思路：

贪心

假设先用完所有优惠券，则只有两种决策：以原价购买未选物品中价格最低的；以最低代价退回当前用的一张优惠券，再以优惠价购买未选物品中优惠价最低的。

回退（反悔）操作可以用堆实现，具体实现：

用三个小根堆h1、h2、h3，分别存储未选物品的原价及编号、未选物品的优惠价及编号、已选物品退回优惠券的代价。

每次比较三个堆的堆顶t1、t2、t3，若当前花费加上min(t1,t2+t3)超过m就break，否则选择两种决策中花费较小的并弹掉堆顶。用bool数组记录每个物品是否被选过。

为了方便实现，最初可以在h3中加入k个0。

---

Code：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, int> node;
const int N = 1e5 + 5;
priority_queue< node, vector<node>, greater<node> > h1, h2;
priority_queue< ll, vector<ll>, greater<ll> > h3;
ll n, k, m, cnt, sum, p[N], c[N];
bool mark[N];
inline void In(ll &num) {
	register char c = getchar();
	for(num = 0; !isdigit(c); c = getchar());
	for(; isdigit(c); num = num * 10 + c - 48, c = getchar());
}
int main() {
	In(n); In(k); In(m);
	for(int i = 1; i <= n; ++i) {
		In(p[i]); In(c[i]);
		h1.push(make_pair(p[i], i));
		h2.push(make_pair(c[i], i));
	}
	for(int i = 1; i <= k; ++i) h3.push(0ll);
	while(!h1.empty()) {
		node t1 = h1.top(), t2 = h2.top();
		ll t3 = h3.top();
		if(mark[t1.second]) {
			h1.pop();
			continue;
		}
		if(mark[t2.second]) {
			h2.pop();
			continue;
		}
		if(t1.first < t2.first + t3) {
			if(sum + t1.first > m) break;
			++cnt;
			h1.pop();
			sum += t1.first;
			mark[t1.second] = true;
		} else {
			if(sum + t2.first + t3 > m) break;
			++cnt;
			h2.pop();
			h3.pop();
			sum += t2.first + t3;
			mark[t2.second] = true;
			h3.push(p[t2.second] - c[t2.second]);
		}
	}
	printf("%lld\n", cnt);
	return 0;
}