题干
给你一个由 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
-10^4 <= nums[i] <= 10^4
解题思路
这道题是典型的滑动窗口 类型的题目。以示例一入手来看题目,k=4,每4个数为一组,由于题目中所说是连续子数组,我们可以将前4个数的和sum算出来,当作最大值max,然后这个可以容纳4个数的窗口向右移动一位,将原本的总和sum减去第一个数,加上下一个数,得到下一组子数组的和,并将其与最大值max作比较,取更大的值做max。以此类推,得到所有连续子数组的和并得到最大的和max。
需要注意,我们得到的和max是int型的,需要通过 max * 1.0 来将其变为double型,确保除法可以得到正确的结果。
AC代码
java
class Solution {
public double findMaxAverage(int[] nums, int k) {
int sum = 0;
int n = nums.length;
for(int i=0;i<k;i++){
sum = sum + nums[i];
}
int max = sum;
for(int i=k;i<n;i++){
sum = sum - nums[i-k] + nums[i];
max = Math.max(max,sum);
}
return 1.0 * max / k;
}
}