【力扣】643.子数组最大平均数--Python实现
【题目描述】 给定 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]。
【解题思路】 采用滑动窗口来解题,求出长度为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
下一篇:
【2023华为OD机试题】分奖金
