给你一个由 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