0
点赞
收藏
分享

微信扫一扫

643. 子数组最大平均数 I


​​643. 子数组最大平均数 I​​

给定 ​​n​​​ 个整数,找出平均数最大且长度为 ​​k​​ 的连续子数组,并输出该最大平均数。

示例:

输入:[1,12,-5,-6,50,3], k = 4
输出:12.75
解释:最大平均数 (12-5-6+50)/4 = 51/4 = 12.75

第一版代码(直接求解: 每K个求一下平均数)

class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
res = sum(nums[:k]) / k
left, right = 1, k
while right < len(nums):
if nums[right] > nums[left - 1]:
res = max(res, sum(nums[left:right + 1])/k)
left += 1
right += 1
return res

643. 子数组最大平均数 I_初始化

第二版代码(k定了, 平均值最大, 那么和就是最大, 不用每次都除以k)

def findMaxAverage(self, nums: List[int], k: int) -> float:
res = sum(nums[:k])
left, right = 1, k
while right < len(nums):
if nums[right] > nums[left - 1]:
res = max(res, sum(nums[left:right + 1]))
left += 1
right += 1
return res / k

比上面的第一版少了一些除法的计算, 但是结果并没有改善反而更差了, 说明实际上改善量很小

643. 子数组最大平均数 I_数据_02

第三版代码(每次都k个数的和, 是比较浪费计算的, 因为每次都只是增加一个数, 减少一个数, 并没有必要每次都重新求和), 效果提升了很多

class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
res = sum(nums[:k])
temp = res
left, right = 1, k
while right < len(nums):
temp = temp + nums[right] - nums[left - 1]
if temp > res:
res = temp
left += 1
right += 1
return res / k

643. 子数组最大平均数 I_LeetCode_03

第四版代码(子数组的长度是k确定的, 那么可以只使用一个变量来进行维护) 并没有改善, 但是其实差别不大

class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
res = sum(nums[:k])
temp = res

# 开始遍历
for i in range(k, len(nums)):
temp = temp - nums[i-k] + nums[i]
if temp > res:
res = temp
return res / k

643. 子数组最大平均数 I_数据_04

第五版代码(考虑两种特殊情况: k=1或k=len(nums)单独表示, 使代码更加清晰) 效果也更好了

class Solution:
def findMaxAverage(self, nums: List[int], k: int) -> float:
size = len(nums)
# 两种特殊情况
if k == 1:
return max(nums)
if k == size:
return sum(nums) / k

# 初始化数据
res = sum(nums[:k])
temp = res

# 开始遍历
for i in range(k, size):
temp = temp - nums[i-k] + nums[i]
if temp > res:
res = temp
return res / k

643. 子数组最大平均数 I_LeetCode_05

举报

相关推荐

0 条评论