【数据结构-堆】力扣1705. 吃苹果的最大数目

有一棵特殊的苹果树，一连 n 天，每天都可以长出若干个苹果。在第 i 天，树上会长出 apples[i] 个苹果，这些苹果将会在 days[i] 天后（也就是说，第 i + days[i] 天时）腐烂，变得无法食用。也可能有那么几天，树上不会长出新的苹果，此时用 apples[i] == 0 且 days[i] == 0 表示。

你打算每天 最多 吃一个苹果来保证营养均衡。注意，你可以在这 n 天之后继续吃苹果。

给你两个长度为 n 的整数数组 days 和 apples ，返回你可以吃掉的苹果的最大数目。

示例 1：
输入：apples = [1,2,3,5,2], days = [3,2,1,4,2]
输出：7
解释：你可以吃掉 7 个苹果：

• 第一天，你吃掉第一天长出来的苹果。
• 第二天，你吃掉一个第二天长出来的苹果。
• 第三天，你吃掉一个第二天长出来的苹果。过了这一天，第三天长出来的苹果就已经腐烂了。
• 第四天到第七天，你吃的都是第四天长出来的苹果。

示例 2：
输入：apples = [3,0,0,0,0,2], days = [3,0,0,0,0,2]
输出：5
解释：你可以吃掉 5 个苹果：

• 第一天到第三天，你吃的都是第一天长出来的苹果。
• 第四天和第五天不吃苹果。
• 第六天和第七天，你吃的都是第六天长出来的苹果。
  

优先队列

class Solution {
public:
    int eatenApples(vector<int>& apples, vector<int>& days) {
        int n = apples.size();
        priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>> q;

        int ans = 0;
        int i = 0;
        for(; i < n; i++){
            while(!q.empty() && (q.top().first <= i)){
                q.pop();
            }

            if(apples[i] != 0 && days[i] != 0){
                q.push({i + days[i], apples[i]});
            }
           
            if(!q.empty()){
                ans++;
                int x = q.top().first;
                int y = q.top().second;
                y--;
                q.pop(); 
                if(y > 0){
                    q.push({x, y});
                }
            }
        }

        while(!q.empty()){
            while(!q.empty() && q.top().first <= i){
                q.pop();
            }

            if(!q.empty()){
                ans++;
                int x = q.top().first;
                int y = q.top().second;
                y--;
                q.pop();
                if(y > 0){
                    q.push({x, y});
                }
            }
            i++;
        }
        return ans;
    }
};

这道题手写一遍过，我的思路就是构造一个优先队列，优先队列的每个二元组的first用来储存腐烂的时间，second用来储存某一批苹果的数量。

首先我们先模拟每天的情况，q的队头是最快腐烂的苹果，我们要将腐烂的苹果从q中弹出。接着我们将新长的苹果加到q中，最后模拟吃掉苹果，将ans加一，然后将q的队头的苹果减少1代表苹果被吃掉。

如果n天后，q队列中还有苹果，那么我们就继续模拟，过程就是之前代码减去新长苹果的那一部分，直到q中没有苹果为止。最后返回ans就是吃苹果的最大数目。