LeetCode 1705 吃苹果的最大数目[贪心 优先队列] HERODING的LeetCode之路

Posted 2022-01-12 HERODING23

解题思路：
这是一道很典型的贪心题目，只要每次吃的都是最先要腐烂的苹果，那么一定能够保证吃到最多的苹果，而且当前只知道目前所拥有的苹果的腐烂时间，之后要收集的苹果和其腐烂时间是不知道的。定义一个优先队列，以腐烂日期最新为排序条件，遍历每一天，先判断是否有苹果可放入，接着丢弃掉腐烂的苹果，最后再吃苹果，由于过了n天还有苹果能吃，所以要符合判断条件，遍历到n天后还有苹果能吃就继续往下走，代码如下：

class Solution 
public:
    int eatenApples(vector<int>& apples, vector<int>& days) 
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
        int count = 0, i = 0;
        int n = apples.size();
        while(i < apples.size() || !q.empty()) 
            // 把当天产的苹果放入队列中
            if(i < n && apples[i] != 0 && days[i] != 0) 
                q.push(i + days[i], apples[i]);
            
            // 丢掉腐烂苹果
            while(!q.empty() && q.top().first <= i) 
                q.pop();
            
            // 吃苹果
            if(!q.empty()) 
                count ++;
                auto [a, b] = q.top();
                q.pop();
                if(b > 1) 
                    q.push(a, b - 1);
                
            
            i ++;
        
        return count;
    
;