Leetcode 643. 子数组最大平均数 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]。

思路
划窗求k个数的和，如果遇到更大的和，替换即可，最后只需返回sum/k。
代码

class Solution {
public:
    double findMaxAverage(vector<int>& nums, int k) {
        double sum = 0;
        double max_sum = 0;
        for(int i = 0; i < k; i++)
        {
            sum += nums[i];
        }
        max_sum = sum;
        for (int i = k; i< nums.size(); i++)
        {
            sum += nums[i] - nums[i-k];
            if( sum > max_sum )
                max_sum = sum;
        }
        return max_sum/k;

    }
};

我同学写的：

class Solution {
public:
    double findMaxAverage(vector<int>& nums, int k) {
        double max_sum = 0;
        double curr_sum = 0;
        for(int i = 0; i < k; ++i)
            curr_sum += nums.at(i);
        max_sum = curr_sum;
        for(int i = k; i < nums.size(); ++i)
        {
            curr_sum += nums.at(i) - nums.at(i - k);
            if(curr_sum > max_sum)
                max_sum = curr_sum;
        }

        return max_sum / k;
    }
};