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 * 10⁴

0 <= apples[i], days[i] <= 2 * 10⁴

只有在 apples[i] = 0 时，days[i] = 0 才成立

解题思路#

第 \(i\) 天应该在 没有腐烂 的苹果中选择 最快要过期 的苹果吃掉。

import heapq

class Solution:
    def eatenApples(self, apples: List[int], days: List[int]) -> int:
        n = len(apples)
        q = []
        i = 0
        ans = 0
        while i < n or q:
            if i < n and apples[i]:  # 第 i 天还能提供苹果
                # [苹果保存的最长时间，苹果数量]
                heapq.heappush(q, [i + days[i] - 1, apples[i]])

            while q:
                e, a = heapq.heappop(q)
                if e >= i:
                    ans += 1
                    a -= 1  # 吃掉一个
                    if a:  # 还有剩余
                        heapq.heappush(q, [e, a])
                    break
            i += 1  # 不管当天有没有苹果吃，时间都增加
        return ans

1705. 吃苹果的最大数目#

问题描述#

解题思路#

1705. 吃苹果的最大数目 #