# 643. 子数组最大平均数 I
力扣原题链接(点我直达) (opens new window)
给定 n
个整数,找出平均数最大且长度为 k
的连续子数组,并输出该最大平均数。
示例 1:
输入: [1,12,-5,-6,50,3], k = 4
输出: 12.75
解释: 最大平均数 (12-5-6+50)/4 = 51/4 = 12.75
1
2
3
2
3
注意:
- 1 <=
k
<=n
<= 30,000。 - 所给数据范围 [-10,000,10,000]。
# 第一版,常规遍历,超时
double aver(vector<int>& res,int k) {
double sum=0;
for (auto& a : res) {
sum += a;
}
return sum / k;
}
double findMaxAverage(vector<int>& nums, int k) {
vector<int> res;
int i = 0;
for (int i = 0; i < k; ++i) {
res.push_back(nums[i]);
}
double average = aver(res,k);
double temp = 0.0;
for ( ; i < nums.size(); ++i) {
res[i% k] = nums[i];
temp = aver(res, k);
average = temp > average ? temp : average;
}
return average;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# 第二版,改进了一下
执行用时 :156 ms, 在所有 cpp 提交中击败了76.31%的用户
内存消耗 :21.1 MB, 在所有 cpp 提交中击败了5.17%的用户
double findMaxAverage(vector<int>& nums, int k) {
vector<int> res;
int i = 0;
for (int i = 0; i < k; ++i) {
res.push_back(nums[i]);
}
int sum = 0;
for (auto& a : res) {
sum += a;
}
int result = sum;
for ( ; i < nums.size(); ++i) {
sum -= res[i % k];
res[i% k] = nums[i];
sum += nums[i];
result = max(sum,result);
}
return result*1.0/k;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# 第三版,改进一点点
执行用时 :164 ms, 在所有 cpp 提交中击败了70.34%的用户
内存消耗 :18.7 MB, 在所有 cpp 提交中击败了7.76%的用户
double findMaxAverage(vector<int>& nums, int k) {
vector<int> res;
res.reserve(k);
int i = 0;
for (; i < k; ++i) {
res.push_back(nums[i]);
}
int sum = accumulate(res.begin(), res.end(), 0);
int result = sum;
for ( ; i < nums.size(); ++i) {
sum -= res[i % k];
res[i% k] = nums[i];
sum += nums[i];
result = max(sum,result);
}
return result*1.0/k;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19