其他分享
首页 > 其他分享> > 643. 子数组最大平均数 I

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
提示:
1 <= k <= n <= 30,000。
所给数据范围 [-10,000,10,000]。

解题

一、滚动数组

用滚动数组来维护maxTotal的更新,减少时间复杂度

二、对maxTotal初始的定义要谨慎

不能粗暴的选取0,因为数组中存在负数,这里选取了第一个平均数

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        maxTotal = total = sum(nums[:k])
        n = len(nums)

        for i in range(k, n):
            total = total - nums[i - k] + nums[i]
            maxTotal = max(maxTotal, total)
        
        return maxTotal / k

在这里插入图片描述

标签:12.75,nums,平均数,643,数组,maxTotal,total
来源: https://blog.csdn.net/qq_51174011/article/details/113638025