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

题解

假设有一组数 nums = [a, b, c, d, e],求 1 - 3 数的总和(b + c + d)

可以先创建一个数组 arr = [0] 初始值为 0

遍历 nums,计算当前值左边所有数的总和,可以得到 arr = [0, a, a+b, a+b+c, a+b+c+d, a+b+c+d+e]

那么 1 - 3数的总和可以算出 arr[4] - arr[1]

任意的 i - j 都可以表示为 arr[j + 1] - arr[i]

在除以个数就可以求出平均数了

  1. /**
  2. * @param {number[]} nums
  3. * @param {number} k
  4. * @return {number}
  5. */
  6. var findMaxAverage = function(nums, k) {
  7. const arr = [0]
  8. let max = -Number.MAX_VALUE
  9. for (let i = 0; i < nums.length; i++) {
  10. arr[i + 1] = arr[i] + nums[i]
  11. }
  12. for (let i = 0; i < nums.length - k + 1; i++) {
  13. const total = arr[i + k] - arr[i]
  14. if (total > max) {
  15. max = total
  16. }
  17. }
  18. return max / k
  19. };

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-average-subarray-i
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。