问题描述
Given an array consisting of n integers, find the contiguous subarray of given length k that has the maximum average value. And you need to output the maximum average value.
Example 1:
Input: [1,12,-5,-6,50,3], k = 4
Output: 12.75
Explanation: Maximum average is (12-5-6+50)/4 = 51/4 = 12.75
Note:
- 1 <= k <= n <= 30,000.
- Elements of the given array will be in the range [-10,000, 10,000].
问题分析
给定一个数组nums[],找出数组中长度等于k的最大子序列的和。解法一是将前i个数的和加起来。然后用第k+i个的和减去第i个的和然后找出最大的。
代码实现
public double findMaxAverage(int[] nums, int k) {
for (int i = 1; i < nums.length; i++) {
nums[i] = nums[i] + nums[i - 1];
}
int maxSum = nums[k - 1];
for (int i = k; i < nums.length; i++) {
int sum = nums[i] - nums[i - k];
if (sum > maxSum) {
maxSum = sum;
}
}
return (double) maxSum / k;
}
算法改进
可以先算出前k个数的和,然后依次加上一个新的值,减去最前面的一个值,统计最大的和就好。这个有点类似于滑动窗口。
代码实现
public double findMaxAverage(int[] nums, int k) {
int sum = 0;
for (int i = 0; i < k; i++) {
sum += nums[i];
}
int res = sum;
for (int i = k; i < nums.length; i++) {
sum = sum + nums[i] - nums[i - k];
// res = Math.max(res, sum);//使用math运行的效率没有比较的效率高
res = res>sum?res:sum;
}
return (double) res / k;
}