Given an array consisting of n integers, find the contiguous subarray of given length k that has the maximum average value. And you need to output the maximum average value.
Example 1:
Input: [1,12,-5,-6,50,3], k = 4
Output: 12.75
Explanation: Maximum average is (12-5-6+50)/4 = 51/4 = 12.75
给出子数组长度k,找出长度为k而且平均值最大的子数组
思路1:
考虑积分数组的思路,就是说每个index处都是0~index这部分的所有元素的和,那么长度为k的子数组的和=积分数组[i] - 积分数组[i-k]
遍历积分数组,找到最大的平均值
//3ms
public double findMaxAverage(int[] nums, int k) {
if(nums == null || nums.length < k) {
return 0;
}
int n = nums.length;
double result = 0;
for(int i = 1; i < n; i++) {
nums[i] += nums[i-1];
}
result = nums[k-1];
for(int i = k; i < n; i++) {
result = Math.max(result, nums[i] - nums[i-k]);
}
return result/k;
}
思路2:
滑动窗口,和上面思路基本一样,只是不用数组来保存所有的和,而是实时更新
还是每一步都求到目前index为止的元素和,但是数组下标超过k的时候,就要每一步多做一件事,就是把k这么长的滑动窗口范围外(左边)的元素减掉
而且下标到达k-1时,就要实时更新子数组和的最大值
和上面思路的不同之处在于,由于是实时更新最大值,result的初始值要初始化到Integer.MIN_VALUE,
注意不能是Double.MIN_VALUE,因为求和过程中是int类型,double类型会越界,得到意料之外的结果
既然初始化成int,那么返回的时候就要做cast to double
//2ms
public double findMaxAverage(int[] nums, int k) {
if(nums == null || nums.length < k || k == 0) {
return 0;
}
int n = nums.length;
int result = Integer.MIN_VALUE;
int sum = 0;
for(int i = 0; i < n; i++) {
if(i >= k) {
sum -= nums[i-k];
}
sum += nums[i];
if(i+1 >= k) {
result = Math.max(result, sum);
}
}
return result/(double)k;
}