AcWing 145. 超市 二叉堆 贪心 优先队列实现

AcWing 145. 超市    

超市里有N件商品，每个商品都有利润$p_i$和过期时间$d_i$,每天只能卖一件商品，过期商品（即当天$d_i$<=0）不能再卖。

求合理安排每天卖的商品的情况下，可以得到的最大收益是多少。

输入格式
输入包含多组测试用例。

每组测试用例，以输入整数N开始，接下里输入N对$p_i和d_i$，分别代表第i件商品的利润和过期时间。

在输入中，数据之间可以自由穿插任意个空格或空行，输入至文件结尾时终止输入，保证数据正确。

输出格式
对于每组产品，输出一个该组的最大收益值。

每个结果占一行。

数据范围
$0≤n≤10000$
$1≤p_i,d_i≤10000$

样例

输入样例：
4  50 2  10 1   20 2   30 1

7  20 1   2 1   10 3  100 2   8 2
   5 20  50 10
输出样例：
80
185

思路：

先将商品按过期时间排序，建立一个小根堆，如果当前商品的过期时间大于堆元素数目，那么直接加入堆。
如果当前商品过期时间=堆元素数目，那么再优化，将此元素的权值与堆顶元素（堆中的最小利润）比较，如果大于，就换掉堆顶元素。

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
priority_queue<int,vector<int> , greater<int> > q;//小根堆 堆顶元素小
const int maxn = 1e5;
pair<int,int> a[maxn];//pair，排序时fisrt为第一关键字
int n;
int main(){
	while(cin>>n){
		if(n==0){//注意特判
		cout<<0<<endl;
		continue;
	}
	for(int i=1;i<=n;i++){
		cin>>a[i].second;//权值
		cin>>a[i].first;//过期天数  因为排序时需要的第一关键字为过期天数 
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++){
		if(a[i].first>q.size()){//如果过期天数大于堆元素数目
			q.push(a[i].second);
		}
	    else if(a[i].first==q.size()&&a[i].second>q.top()){//如果等于，就根据权值优化
	    	q.pop();
	    	q.push(a[i].second);
		}
	}
	ll ans=0;
	while(!q.empty()){//最后留在堆中的就是最终答案
		ans+=q.top();
		q.pop();
	}
	cout<<ans<<endl;
	}
	
}