找到 k 长度的最大平均子数组

在一些数字处理问题中，我们需要寻找一段连续的子数组，该子数组的平均值是这个数组中所有子数组平均值中最大的。在这里，我们提供一种解决这个问题的方法，通过一个长度为k的滑动窗口来得到子数组的平均值，最终，我们返回滑动窗口内平均值最大的子数组。

解决思路

步骤：

1. 初始化一个长度为k的滑动窗口，计算所有元素之和sum，记录下来；
2. 从下一个元素开始，移动滑动窗口，并计算滑动窗口中所有元素之和，得到新的子数组平均值average；
3. 如果average比之前所有求出来的平均值都要大，则更新平均值，并记录下子数组的起始索引start；
4. 最后，返回start处长度为k的子数组。

代码实现

def find_max_average(nums, k):
    start = 0
    sum = 0
    for i in range(k):
        sum += nums[i]
    average = sum / k
    
    for i in range(k, len(nums)):
        sum = sum - nums[i - k] + nums[i]
        new_average = sum / k
        if new_average > average:
            average = new_average
            start = i - k + 1
    
    return nums[start:start + k]

示例

输入：nums = [1,12,-5,-6,50,3], k = 4

输出：[12.75]

解释：长度为4的滑动窗口包括：[12,-5,-6,50]、[1,12,-5,-6]、[12,-5,-6,3]，它们的平均值分别为12.75、0.5和1，显然[12,-5,-6,50]的平均值最大，因此，我们返回该子数组的平均值，即[12.75]。