LeetCode643给定n个整数,找出平均数最大且长度为k的连续子数组,并输出该最大平均数。
其中1<=k<=nums.length<=105。
输入:[1,12,-5,-6,50,3],k = 4 输出:12.75 解释:最大平均数(12-5-6+56)/4 = 51/4 = 12.75
public static double findMaxAverage(int[] nums, int k) {
int len = nums.length;
int widowSum = 0;
if (k > nums.length || nums.length < 1 || k < 1) {
return 0;
}
// 第一步 先求第一个窗口的和
for (int i = 0; i < k; i++) {
widowSum += nums[i];
}
// 第二步 遍历,每次右边增加一个
int res = widowSum;
for (int right = k; right < len; right++) {
widowSum += nums[right] - nums[right - k];
res = Math.max(res, widowSum);
}
return (double) res / k;
}
widowSum += nums[right] - nums[right - k];
这一步实现的就是向右移动一位,那么就需要加上右边的这位,减去最左边的那位。
在这里,很有一种在堆的学习那里的在数组中寻找第K大的元素,那时候的解决方法是先将前K个元素加入堆中,然后再从K开始依次遍历,这里也是一样的。