LeetCode简单题之子数组最大平均数 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 <= 10^5
-10^4 <= nums[i] <= 10 ^4
来源：力扣（LeetCode）

解题思路

  题目给出了子数组的固定长度k，所以这道题采用最简单的滑动窗口就能解决。滑动窗口在滑动的时候更新头部和尾部的指针，并且更新最大总和。

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        i,j=0,k-1
        S=sum(nums[0:k])
        MAX=S
        while j<len(nums)-1:
            S-=nums[i]
            i+=1
            j+=1
            S+=nums[j]
            if S>MAX:
                MAX=S
        return MAX/k