给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例 1:
输入: [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]。
方法1:直接遍历,计算超时
class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
n = len(nums)
if n==0 or n<k or k==0:
return 0
else:
maxN = sum(nums[0:k])/k
for i in range(n-k+1):
if maxN < sum(nums[i:i+k])/k:
maxN = sum(nums[i:i+k])/k
return maxN
方法2:在1的基础上优化,下一轮迭代只需要用上一次的结果减去上一轮数组的第一个值,然后加上接下来的那个值
class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
n = len(nums)
if n==0 or n<k or k==0:
return 0
else:
maxN = sum(nums[0:k])
maxRes = maxN
for i in range(n-k):
maxN = maxN - nums[i] + nums[i+k]
if maxN > maxRes:
maxRes = maxN
return maxRes/k