【力扣】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
经验分享 程序员 微信小程序 职场和发展