题目描述:
给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例:
输入:[1,12,-5,-6,50,3], k = 4
输出:12.75
解释:最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
提示:
- 1 <= k <= n <= 30,000。
- 所给数据范围 [-10,000,10,000]。
解题思路:
首先会直接想到暴力法来解题,将数组中所有可能的子数组结果计算出其平均值,然后找出最大的那个平均值。
class Solution {
public double findMaxAverage(int[] nums, int k) {
int sum=0;
for(int j=0;j<k;j++){
sum+=nums[j];
}
double anum=sum/(k*1.0);
for(int i=1;i<=nums.length-k;i++){
sum=0;
for(int j=i,m=0;m<k;j++,m++){
sum+=nums[j];
}
if(sum/(k*1.0)>anum)
anum=sum/(k*1.0);
}
return anum;
}
}
但这样非常耗时。
优化方案:每次的子数组不再重新计算,采用 减去上一次的首段再加上新的尾部 来计算
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;
}
}