LeetCode刷题记录——第643题(子数组最大平均数)

版权声明:此BLOG为个人BLOG,内容均来自原创及互连网转载。最终目的为记录自己需要的内容或自己的学习感悟,不涉及商业用途,转载请附上原博客。 https://blog.csdn.net/bulo1025/article/details/87856103

题目描述

给定 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)

猜你喜欢

转载自blog.csdn.net/bulo1025/article/details/87856103
今日推荐