给你一个由 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]
在除以个数就可以求出平均数了
/**
* @param {number[]} nums
* @param {number} k
* @return {number}
*/
var findMaxAverage = function(nums, k) {
const arr = [0]
let max = -Number.MAX_VALUE
for (let i = 0; i < nums.length; i++) {
arr[i + 1] = arr[i] + nums[i]
}
for (let i = 0; i < nums.length - k + 1; i++) {
const total = arr[i + k] - arr[i]
if (total > max) {
max = total
}
}
return max / k
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-average-subarray-i
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。