# 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

注意:

  1. 1 <= k <= n <= 30,000。
  2. 所给数据范围 [-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

# 第二版,改进了一下

执行用时 :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

# 第三版,改进一点点

执行用时 :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