给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例 1:
输入: [1,12,-5,-6,50,3], k = 4
输出: 12.75
解释: 最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
注意:
1 <= k <= n <= 30,000。
所给数据范围 [-10,000,10,000]。
解:
维持一个变量max记录 最大“长度为 k 的连续子数组的和”,完成后就用max除以k即可求出结果
代码:
先是一个耗时间但是明了的写法
class Solution {
public double findMaxAverage(int[] nums, int k) {
int max = Integer.MIN_VALUE;
int sum = 0;
//外面循环是“长度为 k 的连续子数组”的头
for (int i = 0; i < nums.length - k + 1; i++)
{
//“长度为 k 的连续子数组”相加
for (int j = i; j < i + k; j++)
{
sum += nums[j];
}
max = Math.max(max, sum);
sum = 0;
}
return (double)max/k;
}
}
滑动窗口
class Solution {
public double findMaxAverage(int[] nums, int k) {
int max = 0, sum = 0;
for(int i = 0; i < k; i++){
sum += nums[i];
}
max = sum;
//逐渐移动向右移动这个窗口大小为k的和
for(int i = k; i < nums.length; i++){
sum = sum - nums[i-k] + nums[i];//减去左端加上右端
max = Math.max(max, sum);
}
return (double)max/k;
}
}