【算法】滑动窗口(细节探究,易错解析)5.21
接上节
需求:
给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。
请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。
示例 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
方式一
使用双循环
class Solution {public double findMaxAverage(int[] nums, int k) {double ans=Integer.MIN_VALUE;double avg=0;double sum=0;if(nums==null||nums.length<k){return 0;}for(int j=0;j<=nums.length-k;j++){sum=0.0;for(int i=j;i<k+j;i++){sum+=nums[i];}avg=sum/k;ans=Math.max(ans,avg);}return ans;}
}
易错点:
①ans代表平均数最大值,如果赋值为0,当数组为复数时,会返回0报错。
所以ans应该赋值为Integer.MIN_VALUE(代表最小值)。
②sum要重置,如果不重置,每次都会累加上一组循环的值。
③j的循环要保证包括k边界,
j<=nums.length-k或j-k+1<nums.length
④需要判断数组长度是否大于k,否则返回0。
⑤double类型也要注意
方式二
滑动窗口
class Solution {public double findMaxAverage(int[] nums, int k) {double ans=Integer.MIN_VALUE;double sum=0;for(int i=0;i<nums.length;i++){sum+=nums[i];if(i<k-1){continue;}ans=Math.max(ans,sum);sum-=nums[i-k+1];}return ans/k;}
}
说明:
①continue的作用,如果不足k个元素,就不会执行下面的操作,保证满足数组k的长度。
②更新最大值
③sum-=nums[i-j+1]减去出去的元素。
方式三
滑动窗口第二种形式
class Solution {public double findMaxAverage(int[] nums, int k) {int sum = 0;int n = nums.length;for (int i = 0; i < k; i++) {sum += nums[i];}int maxSum = sum;for (int i = k; i < n; i++) {sum = sum - nums[i - k] + nums[i];maxSum = Math.max(maxSum, sum);}return 1.0 * maxSum / k;}
}
说明:
①先得到第一组数组的和
②通过遍历更新数组