主要知识点
求平均啊,大家应该都会,不会可以直接看英雄哥的专栏 😄
习题
1. Leetcode 643. 子数组最大平均数 I
题目描述
给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。
请你找出平均数最大且 长度为 k 的连续子数组,并输出该最大平均数。
任何误差小于 10-5 的答案都将被视为正确答案。
提示: n = = n u m s . l e n g t h n == nums.length n==nums.length
1 ≤ k ≤ n ≤ 105 1 \leq k \leq n \leq 105 1≤k≤n≤105
− 104 ≤ n u m s [ i ] ≤ 104 -104 \leq nums[i] \leq 104 −104≤nums[i]≤104
初见
为了计算连续 k k k 个数的平均数,我们可以使用双指针的方式,在区间头与区间尾维持两个指针,当计算下一个数时,区间尾指针右移,区间头指针同时右移,保持区间长度 k k k 不变,更新平均数累加和并计算新区间的平均数,与当前最大平均数进行比较后判断是否更新:
double findMaxAverage(vector<int>& nums, int k) {
// 使用双指针维持计算区间
int l(0), r(0);
double count(0);
double aveResult(0);
for (int i = 0; i < nums.size(); ++i) {
if ((r - l + 1) < k) {
count += nums[i]; // 不足 k 个数时累加
} else if ((r - l + 1) == k) {
// 计算平均数并重置和
count += nums[i];
// std::cout << l << " " << r << " " << count << std::endl;
double curAve = count / k;
if (l == 0) { // 第一次初始化
aveResult = curAve;
}
if (curAve > aveResult) {
aveResult = curAve;
}
count -= nums[l];
l += 1; // 区间右移
}
r += 1;
}
return aveResult;
}
总结
- 对于涉及到区间计数的题目使用双指针大部分都可以有效解决