leetcode-643. 子数组最大平均数 I
一、问题描述:
给定 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]。
二、代码和思路
采用向后遍历数组的方法,不断增加nums[i,k]的最后一个元素以及减少前一个元素,找最大的max_sum然后记录
class Solution(object):
def findMaxAverage(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: float
"""
i,sum_now,max_sum,n,=0,sum(nums[0:k]),sum(nums[0:k]),len(nums)
while i<n-k:
i += 1
sum_now += (nums[i+k-1]-nums[i-1])
if sum_now>max_sum:
max_sum=sum_now
return float(max_sum)/k
三、运行结果