643. 子数组最大平均数 I
给定 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]。
第一版,常规遍历,超时
doubleaver(vector<int>&res,intk) {
doublesum=0;
for (auto&a : res) {
sum+=a;
}
returnsum/k;
}
doublefindMaxAverage(vector<int>&nums, intk) {
vector<int>res;
inti=0;
for (inti=0; i<k; ++i) {
res.push_back(nums[i]);
}
double average=aver(res,k);
doubletemp=0.0;
for ( ; i<nums.size(); ++i) {
res[i%k] =nums[i];
temp=aver(res, k);
average=temp>average?temp : average;
}
returnaverage;
}
第二版,改进了一下
执行用时 :156 ms, 在所有 cpp 提交中击败了76.31%的用户
内存消耗 :21.1 MB, 在所有 cpp 提交中击败了5.17%的用户
doublefindMaxAverage(vector<int>&nums, intk) {
vector<int>res;
inti=0;
for (inti=0; i<k; ++i) {
res.push_back(nums[i]);
}
intsum=0;
for (auto&a : res) {
sum+=a;
}
intresult=sum;
for ( ; i<nums.size(); ++i) {
sum-=res[i%k];
res[i%k] =nums[i];
sum+=nums[i];
result=max(sum,result);
}
returnresult*1.0/k;
}
第三版,改进一点点
执行用时 :164 ms, 在所有 cpp 提交中击败了70.34%的用户
内存消耗 :18.7 MB, 在所有 cpp 提交中击败了7.76%的用户
doublefindMaxAverage(vector<int>&nums, intk) {
vector<int>res;
res.reserve(k);
inti=0;
for (; i<k; ++i) {
res.push_back(nums[i]);
}
intsum=accumulate(res.begin(), res.end(), 0);
intresult=sum;
for ( ; i<nums.size(); ++i) {
sum-=res[i%k];
res[i%k] =nums[i];
sum+=nums[i];
result=max(sum,result);
}
returnresult*1.0/k;
}