题目
给定 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]。
思路
划窗求k个数的和,如果遇到更大的和,替换即可,最后只需返回sum/k
。
代码
class Solution {
public:
double findMaxAverage(vector<int>& nums, int k) {
double sum = 0;
double max_sum = 0;
for(int i = 0; i < k; i++)
{
sum += nums[i];
}
max_sum = sum;
for (int i = k; i< nums.size(); i++)
{
sum += nums[i] - nums[i-k];
if( sum > max_sum )
max_sum = sum;
}
return max_sum/k;
}
};
我同学写的:
class Solution {
public:
double findMaxAverage(vector<int>& nums, int k) {
double max_sum = 0;
double curr_sum = 0;
for(int i = 0; i < k; ++i)
curr_sum += nums.at(i);
max_sum = curr_sum;
for(int i = k; i < nums.size(); ++i)
{
curr_sum += nums.at(i) - nums.at(i - k);
if(curr_sum > max_sum)
max_sum = curr_sum;
}
return max_sum / k;
}
};