题目:
给定 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]。
Python代码1:(未AC)
class Solution:
def findMaxAverage(self, nums, k):
maximum = -float('inf')
for i in range(len(nums)-k+1):
maximum = max(maximum,sum(nums[i:i+k]))
return maximum/k
python代码2:
class Solution:
def findMaxAverage(self, nums, k):
maximum = -float('inf')
res = 0
for i in range(len(nums)):
res += nums[i]
if i >= k:
res -= nums[i-k]
if i >= k-1:
maximum = max(maximum, res/k)
return maximum
心得:代码1比较简洁,但是在测试比较长的案例时没有通过(考虑是由于sum的时间复杂度偏大),于是考虑滑动窗口机制,如以上代码2,成功解决问题。