【题目描述】
给定 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的滑动窗口的最大值/k即为所求。用Python实现的代码如下:
class Solution(object):
def findMaxAverage(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: float
"""
slide_window = []
MIN = -3e8 - 1
tmp, res = 0, MIN
for i in range(len(nums)):
if len(slide_window)<k:
slide_window.append(nums[i])
tmp += nums[i]
if len(slide_window)==k:
if tmp>res:
res = tmp
first_item = slide_window.pop(0)
tmp -= first_item
return 1.0*res/k