力扣 leetcode 643. 子数组最大平均数 I （python）

Topic

给定 n 个整数，找出平均数最大且长度为 k 的连续子数组，并输出该最大平均数。

Example

输入：[1,12,-5,-6,50,3], k = 4
输出：12.75
解释：最大平均数 (12-5-6+50)/4 = 51/4 = 12.75

Tips

1 <= k <= n <= 30,000。
所给数据范围 [-10,000，10,000]。

Solution

很明显这也是一道滑动窗口的题目
平均数较大则总数一定更大
为了增加运行效率仅需在遍历窗口时计算总数
最后求平均数

首先计算第一个窗口的总数为total
同时将total赋值给结果res

继续遍历后面的值
同时total中减去窗口左侧删除的数字，加上右侧进入窗口的数字
若total比res大
则将total赋值给res

最后将res除以k求得最后的平均数

Code

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        total = sum(nums[:k])
        res = total
        n = len(nums)

        for i in range(k, n):
            total = total - nums[i - k] + nums[i]
            res = max(total, res)
        
        return res / k

Result