LeetCode：子数组最大平均数 I

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;
    }
};