吃苹果的最大数目

2021-12-24  本文已影响0人  xialu

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-number-of-eaten-apples

题目描述:

有一棵特殊的苹果树,一连 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(int[] a, int[] d) {
       PriorityQueue<int[]> q = new PriorityQueue<>((c,b) -> c[0]-b[0]);
       int n = a.length;
       int time = 0;
       int ans = 0;
       while (time < n || !q.isEmpty()) {
           if (time < n && a[time] > 0) q.add(new int[]{time + d[time] - 1, a[time]});
           while (!q.isEmpty() && q.peek()[0] < time) q.poll();
           if (!q.isEmpty()) {
               int[] cur = q.poll();
               if (--cur[1] > 0 && cur[0] > time) q.add(cur);
               ans++;
           }
           time++;
       }
       return ans;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读