题目:
给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。
示例 1:
输入: [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) {
double ave = 0.0,sum = 0.0;
double maxAve = -10000;
int len = nums.length;
if(len <= k){
for(int i=0;i<len;i++){
sum += nums[i];
}
maxAve = sum/len;
return maxAve;
}
for(int i=0;i<=len-k;i++){
for(int j=i;j<i+k;j++){
sum += nums[j];
}
ave = sum/k;
if(ave > maxAve){
maxAve = ave;
}
sum = 0.0;
}
return maxAve;
}
}
效率:
代码实现②:
class Solution {
public:
double findMaxAverage(vector<int>& nums, int k) {
double sum = 0.0,result = 0.0;
for(int i=0;i<k;i++){
sum += nums[i];
}
result = sum;
for(int i=1;i+k-1<nums.size();i++){
result = result + nums[i+k-1]-nums[i-1];
if(result > sum){
sum = result;
}
}
return sum/k;
}
};
效率: