LeetCode刷题记录——第643题（子数组最大平均数）

题目描述

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

示例 1:

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

思路分析

• 本题考察的就是一个固定窗口求和，然后窗口的移动
• 关键步骤为： res = res - nums[i] + nums[i+k]，表示窗口整体的后移一个单位
• 最后返回的时候，分母为什么要加一个0.0呢？是因为为了保证运算结果保留了小数，否则只会是整数，结果就不对了

代码示例

class Solution(object):
    def findMaxAverage(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: float
        """
        res = sum(nums[0:k])
        max_ = res
        for i in range(len(nums)-k):
            if len(nums[i:i+k]) != k:
                break
            res = res - nums[i] + nums[i+k]
            if res > max_:
                max_ = res
        return max_ / (k+0.0)