643. 子数组最大平均数 I

643. 子数组最大平均数 I

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

示例：

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

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

解题

一、滚动数组

用滚动数组来维护maxTotal的更新，减少时间复杂度

二、对maxTotal初始的定义要谨慎

不能粗暴的选取0，因为数组中存在负数，这里选取了第一个平均数

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

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