【力扣】1705. 吃苹果的最大数目

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 个苹果：
- 第一天到第三天，你吃的都是第一天长出来的苹果。
- 第四天和第五天不吃苹果。
- 第六天和第七天，你吃的都是第六天长出来的苹果。

提示：

• apples.length == n
• days.length == n
• $1<=n<=2\ast 10^4$
• $0<=apples[i],days[i]<=2\ast 10^4$
• 只有在 apples[i] = 0 时，days[i] = 0 才成立

题解

方法一：贪心 + 优先队列

class Solution {
    public int eatenApples(int[] apples, int[] days) {
        int ans = 0;
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>((a, b) -> a[0] - b[0]);  // a[0]是腐烂日期，a[1]是苹果个数
        int n = apples.length;
        int i = 0;
        while (i < n) {   // i是时间
            while (!pq.isEmpty() && pq.peek()[0] <= i) {  // 队首已经腐烂了，出队
                pq.poll();
            }
            int rottenDay = i + days[i];
            int count = apples[i];
            if (count > 0) {
                pq.offer(new int[]{rottenDay, count});  // 当前苹果及其腐烂时间入队
            }
            if (!pq.isEmpty()) {
                int[] arr = pq.peek();
                arr[1]--;
                if (arr[1] == 0) {   //  苹果吃完了，队首出队
                    pq.poll();
                }
                ans++;
            }
            i++;
        }
        while (!pq.isEmpty()) {  //  没有新生产的苹果了，计算队列中的苹果
            while (!pq.isEmpty() && pq.peek()[0] <= i) {
                pq.poll();
            }
            if (pq.isEmpty()) {
                break;
            }
            int[] arr = pq.poll();
            int curr = Math.min(arr[0] - i, arr[1]);  // 当前苹果可以吃curr天
            ans += curr;   // 结果+curr
            i += curr;   // 时间+curr
        }
        return ans;
    }
}

/**
贪心：每天应该吃剩余保质期最短的苹果。
用一个优先队列，保存(腐烂日期, 腐烂的数量)，每次取队首就是剩余保质期最短的。到腐烂日期或者腐烂数量归零（提前吃完了）的节点出队。
 */
class Solution {
    public int eatenApples(int[] apples, int[] days) {
        int ans = 0;
        Queue<int[]> queue = new PriorityQueue<>((x1, x2) -> x1[0]-x2[0]);
        int d = 0;
        while (d < apples.length || !queue.isEmpty()){
            if (d < apples.length && apples[d] > 0) queue.add(new int[]{d+days[d], apples[d]});       // 新结出的苹果入队
            while (!queue.isEmpty() && (queue.peek()[0] <= d || queue.peek()[1] == 0)) queue.poll();  // 到腐烂日期或者腐烂数量归零
            if (!queue.isEmpty()) {     // 吃苹果
                ++ans;
                --queue.peek()[1];
            }
            ++d;
        }
        return ans;
    }
}