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开始依次遍历,这里也是一样的。