问题描述
给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例:
输入:[1,12,-5,-6,50,3], k = 4
输出:12.75
解释:最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
提示:
1 <= k <= n <= 30,000。
所给数据范围 [-10,000,10,000]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-average-subarray-i
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析及代码实现
先定义一个固定的窗口,然后求出前k数之和和平均数,然后将这个和每次减去开头的数并加上后面的数,求当前窗口的平均数,然后和之前的进行比较大小,留下大的,最后返回平均值
class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
total = sum(nums[:k])
res = total/k
for i in range(k, len(nums)):
total =total - nums[i-k] + nums[i]
res = max(res, total/k)
return res