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

}

第二版,改进了一下

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

}

第三版,改进一点点

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

}