目录

LeeCode题库第643题

LeeCode题库第643题

项目场景:

给你一个由 n 个元素组成的整数数组 nums 和一个整数 k

请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。

任何误差小于 10-5 的答案都将被视为正确答案。

示例 1:

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

示例 2:

输入:nums = [5], k = 1
输出:5.00000

提示:

  • n == nums.length
  • 1 <= k <= n <= 105
  • -104 <= nums[i] <= 104

问题描述

这题同样也是滑动窗口的思想,cur为当前k个数的和,ans为最大值。利用滑动窗口的方法可以将子数组中保持在k个,这样只需每次移入移出即可,最终返回最大的ans即可。

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        ans=float("-inf")
        cur=0
        for r,num in enumerate(nums):
            cur+=num
            if r<k-1:
                continue
            ans=max(ans,cur/k)
            cur-=nums[r-k+1]
        return ans

        

以上为本篇文章的全部内容,感谢你抽出宝贵的时间阅读这篇文章。如果你有任何疑问或建议,欢迎在评论区留言,我们一起交流进步。愿你的代码之路越走越顺,生活充满阳光!