LeetCode：1705. 吃苹果的最大数目（优先级队列 + 贪心 Java）

目录

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 * 104
• 0 <= apples[i], days[i] <= 2 * 104
• 只有在 apples[i] = 0 时，days[i] = 0 才成立

实现代码与解析：

优先级队列 + 贪心

class Solution {
    public int eatenApples(int[] apples, int[] days) {


        PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> a[0] - b[0]);

        int n = apples.length;

        int res = 0;
        int ma = 0;
        for (int i = 0; i < n || i < ma ;i ++) {
            if (i < n) {
                pq.add(new int[]{i + days[i], apples[i]});
                ma = Math.max(ma, i + days[i]);
            }
            
            while (!pq.isEmpty() && (pq.peek()[0] <= i || pq.peek()[1] <= 0)) {
                pq.poll();
            }
            if (!pq.isEmpty()) {
                int[] cur = pq.peek();
                pq.poll();
                if (--cur[1] != 0) {
                    pq.add(cur);
                }
                res++;
            }
        }
        return res;
    }
}

原理思路：

        优先级队列维护最早过期的苹果，贪心思路就是每次吃最早过期的水果，最后一定可以吃的最多。注意如果过期或者某一批苹果被吃光时，从队列中poll出来。不过我这里取出某一批，如果吃完后变为0了也不会放进去，两个判断留一个就行。

        注意天数结束任然可以继续吃，所以我们要遍历到最后过期的天数，而不是最后长出苹果的天数。