给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例:
输入:[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]。
解题思路:
滑动窗口思想。
1.初始化将滑动窗口压满,取得第一个滑动窗口的目标值
2.继续滑动窗口,每往前滑动一次,需要删除一个和添加一个元素
不用滑动窗口,直接计算子数组和会超时。
以下为代码模板:
class Solution {
public double findMaxAverage(int[] nums, int k) {
double ans = 0, sum = 0;
for (int i = 0; i < k; i++) sum += nums[i];
ans = sum / k;
for (int i = k; i < nums.length; i++) {
sum = sum + nums[i] - nums[i - k]; // int add = nums[i], del = nums[i - k]; 从子数组最后一个元素来看
ans = Math.max(ans, sum / k);
}
return ans;
}
}
- 时间复杂度:每个元素最多滑入和滑出窗口一次。复杂度为 O(n)O(n)
- 空间复杂度:O(1)O(1)
class Solution {
public double findMaxAverage(int[] nums, int k) {
double max = 0, sum = 0;
for (int i = 0; i < k; i++) max += nums[i];
sum = max;
for (int i = k; i < nums.length; i++) {
sum = sum + nums[i] - nums[i - k]; // int add = nums[i], del = nums[i - k];
max = Math.max(sum, max);
}
return max / k ;
}
}
class Solution {
public double findMaxAverage(int[] nums, int k) {
double max = 0 ;
for(int i=0;i<k;i++){
max += nums[i];
}
double sum = max;
for(int i=1;i<=nums.length-k;i++){
sum -= nums[i-1];
sum += nums[i+k-1];
max = Math.max(sum,max);
}
return max / k;
}
}