给定 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]。
例如 , 输入 [0,1,2,3,4,5,6] , k = 4
有三个关键变量sum,max,x
sum 初始化为最开始的k个数,sum = 0+1+2+3,以后每次迭代都是减去第一个数,加上后面一个数
max用于记录最大的和
x 用于计算本次的k个数的和
代码如下:
class Solution:
def findMaxAverage(self, nums,k):
"""
:type nums: List[int]
:type k: int
:rtype: float
"""
lens = len(nums)
sum = 0
max = 0
for i in range(k): #初始化
sum = sum + nums[i]
max = sum
for i in range(1,lens-k+1):
x = sum + nums[k+i-1] - nums[i-1]
if x>max:
max = x
sum = x
return max/k
class Solution:
def findMaxAverage(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: float
"""
cur = sum(nums[:k]) #初始化,计算前k项
max_val = cur
for i in range(1, len(nums)-k+1):
cur += nums[i+k-1]-nums[i-1]
if cur > max_val:
max_val = cur
return max_val/k
这段代码的思路和我的一样,但是比我快,主要是以下几点: