Leetcode 1705. 吃苹果的最大数目(medium)

题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-number-of-eaten-apples
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

1. 原始解法

使用一个列表food来存储信息，列表的每个元素是[i + days[i], apples[i]]

前n天：

        (1). 收获苹果：当某一天的apples[i]不为0时，才向food添加元素

food = []
for i in range(n):  
    # at day i, we first harvest apples
    if apples[i] != 0:
        food.append([i + days[i], apples[i]])

        (2).对food进行排序，依据为i + days[i]：排序是为了吃最快腐烂的苹果

# then we need to sort the list based on i+days[i]
food.sort(key=lambda x: x[0])

        (3).如果有苹果，则吃一个

# now we have to find out whether we have an apple to eat
if len(food) != 0:
    food[0][1] -= 1
    count += 1

        (4).丢弃所有将要腐烂的苹果（以及被吃完的food[0]）

# if we ate all apples from food[0], drop it
if len(food) != 0:
    if food[0][1] == 0:
        food.pop(0)
            
# at the end of day, we need to throw away all rotten apples
if len(food) != 0:
    while (food[0][0] == i + 1):
        food.pop(0)

        if len(food) == 0:
            break

后续：

        每一天吃一个苹果，并丢弃过期的。这里我使用了for循环，所以需要先确定最后一批苹果腐烂的那一天。如果前n天吃掉了所有水果，则无需处理后续部分。

# max_day represents the last day we have an apple
if len(food) > 0:
    max_day = food[-1][0]
else:
    max_