643子数组最大平均数I

给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。

请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。

任何误差小于 10-5 的答案都将被视为正确答案。

示例 1:

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

示例 2:

输入:nums = [5], k = 1
输出:5.00000

提示:

  • n == nums.length
  • 1 <= k <= n <= 105
  • -104 <= nums[i] <= 104

Solution 1

class Solution:
    def findMaxAverage(self, nums: [int], k: int) -> float:
        n = len(nums)
        sum_k = 0
        max_sum_k = -1000*k
        for i in range(n):
            sum_k += nums[i]
            if i >= k-1:
                max_sum_k = max(max_sum_k, sum_k)
                sum_k -= nums[i-k+1]
        return max_sum_k / k