思路:将苹果的腐烂日期、个数存入优先队列,类型为pair,根据腐烂日期升序排序,我们只需在每天对优先队列进行维护,包括移除腐烂的苹果,加入新生的苹果,吃苹果。
class Solution {
public:
int eatenApples(vector<int>& apples, vector<int>& days) {
int res=0;
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>save;
//注意第n天后还可以吃,所以条件加上队列不为空
for(int i=0;i<apples.size()||!save.empty();i++){
//移除腐烂的苹果
while(!save.empty()&&save.top().first==i){
save.pop();
}
//加入第i天新生的苹果
if(i<apples.size()&&apples[i]!=0){
save.push(make_pair(i+days[i],apples[i]));
}
//吃苹果,注意将没有吃完的苹果重新加入优先队列
if(!save.empty()){
pair<int,int> temp=save.top();
save.pop();
res++;
temp.second--;
if(temp.second>0){
save.push(make_pair(temp.first,temp.second));
}
}
}
return res;
}
};