【leetcode】子数组的最大平均数

子数组的最大平均数

问题描述

简单解法

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        sum=0
        allres=[]
        for i in range(len(nums)-k+1):
            for m in nums[i:i+k]:
                sum+=m
            allres.append(sum)
            sum=0
        return max(allres)/k

移动窗口求每次的平均值：时间复杂度大

优化解法

滑动窗口时不要每次都重新求和，而是减去滑出的值再加上滑入的值，可以大大减小时间复杂度。

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        sumu=0
        for i in range(k):
            sumu+=nums[i] 
        allres=[]
        allres.append(sumu)
        for i in range(1,len(nums)-k+1):
            sumu-=nums[i-1]
            sumu+=nums[i+k-1]
            allres.append(sumu)
        print(allres)
        return max(allres)/k