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

第一版代码(直接求解: 每K个求一下平均数)

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        res = sum(nums[:k]) / k
        left, right = 1, k
        while right < len(nums):
            if nums[right] > nums[left - 1]:
                res = max(res, sum(nums[left:right + 1])/k)
            left += 1
            right += 1
        return res 

第二版代码(k定了, 平均值最大, 那么和就是最大, 不用每次都除以k)

    def findMaxAverage(self, nums: List[int], k: int) -> float:
        res = sum(nums[:k])
        left, right = 1, k
        while right < len(nums):
            if nums[right] > nums[left - 1]:
                res = max(res, sum(nums[left:right + 1]))
            left += 1
            right += 1
        return res / k

比上面的第一版少了一些除法的计算, 但是结果并没有改善反而更差了, 说明实际上改善量很小

第三版代码(每次都k个数的和, 是比较浪费计算的, 因为每次都只是增加一个数, 减少一个数, 并没有必要每次都重新求和), 效果提升了很多

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        res = sum(nums[:k])
        temp = res
        left, right = 1, k
        while right < len(nums):
            temp = temp + nums[right] - nums[left - 1]
            if temp > res:
                res = temp
            left += 1
            right += 1
        return res / k

第四版代码(子数组的长度是k确定的, 那么可以只使用一个变量来进行维护) 并没有改善, 但是其实差别不大

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

        # 开始遍历
        for i in range(k, len(nums)):
            temp = temp - nums[i-k] + nums[i]
            if temp > res:
                res = temp
        return res / k

第五版代码(考虑两种特殊情况: k=1或k=len(nums)单独表示, 使代码更加清晰) 效果也更好了

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        size = len(nums)
        # 两种特殊情况
        if k == 1:
            return max(nums)
        if k == size:
            return sum(nums) / k
        
        # 初始化数据
        res = sum(nums[:k])
        temp = res

        # 开始遍历
        for i in range(k, size):
            temp = temp - nums[i-k] + nums[i]
            if temp > res:
                res = temp
        return res / k