来自LEETCODE643:
给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。
请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。
任何误差小于 10-5 的答案都将被视为正确答案。
示例 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
提示:
- n == nums.length
- 1 <= k <= n <= 105
- -104 <= nums[i] <= 104
分析如下:
💡 Tips:此题是典型的滑动窗口问题,而大小也已经规定好了,就是K。那我们选择先只读取K个然后让窗口逐步向前走就得到了解决。既然这样的话,我们再来看看对应的图示吧:
我们选择去建立两个索引,即left和right,保持(left,right)之间一共有三个元素,然后我们选择一边遍历,一边去寻找,每当改变一次就去标记一下当前区间的最大值就好了。
代码块
💡 Tips:废话不多说我们直接上代码
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;
}
为了方便读者阅读,我们再来逐步解读一下代码:
💡 Tips:
这段代码是用于找到一个数组中长度为k的连续子数组的平均值的最大值。函数接受两个参数,一个是整数数组nums,另一个是整数k。首先,代码会判断k是否大于数组长度、数组长度是否小于1以及k是否小于1,如果满足其中一个条件,就会返回0。
接下来,代码会计算数组中第一个长度为k的子数组的和,使用一个循环来累加前k个元素的值。
然后,代码会遍历数组,每次将右边界右移一个位置,左边界左移一个位置,重新计算窗口内元素的和,并更新最大值。
最后,代码会返回最大值除以k的结果作为平均值的最大值。