版权声明:此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)