leetcode-每日一题2021.12.24 吃苹果的最大数目

题目

力扣

思路 贪心 + 优先队列

根据苹果的腐烂日期排序，把[腐烂日期，苹果个数]二元组放入优先队列中。对每一天的苹果进行遍历：先判断队首的苹果是否腐烂，如果腐烂就pop出。再把当前的[腐烂日期，苹果个数]放入优先队列。如果优先队列不为空，就取出队首元素，将苹果个数-1.如果减完之后，苹果个数依然不是0，就把这个二元组再放入优先队列中。遍历完后，如果优先队列还不为空，就先判断里面的苹果是否腐烂，如果还有没腐烂的，就把优先队列中苹果取出，取腐烂时间-当前时间和苹果个数的最小值，加到结果中，当前时间也做相应的推进。

代码

typedef pair<int,int> pii;
class Solution {
public:
    int eatenApples(vector<int>& apples, vector<int>& days) {
        priority_queue<pii,vector<pii>,greater<pii>> pq;
        int n=apples.size(),time=0,cnt=0;
        while(time<n || !pq.empty()){
            //把当前时间长出的苹果放入优先队列
            if(time<n && apples[time]>0)
                pq.emplace(time+days[time]-1,apples[time]);
            //把腐烂的苹果pop出优先队列
            while(!pq.empty() && pq.top().first<time)
                pq.pop();
            //从优先队列中拿出苹果
            if(!pq.empty()){
                pii cur=pq.top();
                pq.pop();
                if(--cur.second>0) pq.push(cur);
                cnt++;
            }
            time++;
        }
        return cnt;
    }
};