LeetCode:子数组最大平均数 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]。
解题思路
dp[i]是前i个数字的和,然后从前往后当遇到索引i大于k的时候,计算i-k到i之间的和最大即可。
由于dp数组保存了前面的和,i-k到i区间的和等于dp[i] - dp[i-k]。
时间复杂度O(n)!
class Solution {
public:
double findMaxAverage(vector<int>& nums, int k) {
int dp[30002] = {0};
if(nums.size() == 1)
return nums[0];
dp[1] = nums[0];
double max_sum = INT_MIN;
if( k == 1)
max_sum = nums[0];
for( int i = 2; i <= nums.size(); i++)
{
dp[i] = dp[i-1] + nums[i-1];
if( i >= k && dp[i] - dp[i-k] > max_sum)
max_sum = dp[i] - dp[i-k];
}
return max_sum/k;
}
};