本文共 1770 字，大约阅读时间需要 5 分钟。

你的好友是一位健身爱好者。前段日子，他给自己制定了一份健身计划。现在想请你帮他评估一下这份计划是否合理。

他会有一份计划消耗的卡路里表，其中 calories[i] 给出了你的这位好友在第 i 天需要消耗的卡路里总量。

为了更好地评估这份计划，对于卡路里表中的每一天，你都需要计算他 「这一天以及之后的连续几天」 （共 k 天）内消耗的总卡路里 T：

• 如果 T < lower，那么这份计划相对糟糕，并失去 1 分；
• 如果 T > upper，那么这份计划相对优秀，并获得 1 分；
• 否则，这份计划普普通通，分值不做变动。

请返回统计完所有 calories.length 天后得到的总分作为评估结果。

注意：总分可能是负数。

示例 1：

输入：calories = [1,2,3,4,5], k = 1, lower = 3, upper = 3输出：0解释：calories[0], calories[1] < lower 而 calories[3], calories[4] > upper, 总分 = 0.

示例 2：

输入：calories = [3,2], k = 2, lower = 0, upper = 1输出：1解释：calories[0] + calories[1] > upper, 总分 = 1.

示例 3：

输入：calories = [6,5,0,0], k = 2, lower = 1, upper = 5输出：0解释：calories[0] + calories[1] > upper, calories[2] + calories[3] < lower, 总分 = 0.

提示：

• 1 <= k <= calories.length <= 10^5
• 0 <= calories[i] <= 20000
• 0 <= lower <= upper

解题思路

这个问题非常简单，计算所有区间长度为k的区间的和，然后统计lower和upper的次数即可。我们使用t存储区间长度为k的区间和，当我们遍历的元素位置大于k-1的时候，我们此时就得到了第一个长度为k的区间

0 1 ... k-1 k          ↑

所以此时我们需要判断此时的t满足哪个条件。接着遍历到k个元素

0 1 ... k-1 k   ↑         ↑

此时为了维护区间长度，我们需要将第0个元素去除。

class Solution:    def dietPlanPerformance(self, calories: List[int], k: int, lower: int, upper: int) -> int:        data, t, res = [], 0, 0        for i, v in enumerate(calories):            t += v            if i >= k - 1:                if t < lower:                    res -= 1                elif t > upper:                    res += 1                t -= calories[i-k+1]        return res

一个更加简洁的写法

class Solution:    def dietPlanPerformance(self, calories: List[int], k: int, lower: int, upper: int) -> int:        T = sum(calories[:k])        res = (T > upper) - (T < lower)        for i in range(k, len(calories)):            T += calories[i] - calories[i - k]            res += (T > upper) - (T < lower)        return res

我将该问题的其他语言版本添加到了我的

如有问题，希望大家指出！！！

转载地址：https://coordinate.blog.csdn.net/article/details/100529756 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！