力扣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

思路:

滑动窗口法,从1到n-k+1sum(nums[:k]),减去nums[i-1],加上nums[i+k-1],比较当前最大值和最新值,取max = (ans,current_sum)

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        # 初始化 ans 为前 k 个元素的和
        n = len(nums)
        ans = sum(nums[:k])
        current_sum = ans
        
        # 使用滑动窗口法
        for i in range(1, n - k + 1):
            # 从窗口中移除前一个元素,添加新的元素
            current_sum = current_sum - nums[i - 1] + nums[i + k - 1]
            # 更新最大和
            ans = max(ans, current_sum)
        
        # 返回最大平均值
        return ans / k

上一篇:java-day7-练习六:数字加密


下一篇:【优选算法篇】双指针的优雅舞步:C++ 算法世界的浪漫探索