leetcode-643.子数组最大平均数 I

最新推荐文章于 2024-10-31 16:16:13 发布

wwiiinn

最新推荐文章于 2024-10-31 16:16:13 发布

阅读量137

点赞数 2

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/zjinyu_/article/details/142346110

版权

算法题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述

给你一个由 n 个元素组成的整数数组 nums 和一个整数 k 。
请你找出平均数最大且长度为 k 的连续子数组，并输出该最大平均数。
任何误差小于 10^-5 的答案都将被视为正确答案。

滑动窗口解决该问题。
代码1(没考虑小于0的输入，错误版本)：

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

时间复杂度:O(n)
空间复杂度:O(1)

代码2(以为DBL_MIN是Double类型的最小值，错误版本)：

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

时间复杂度:O(n)
空间复杂度:O(1)

Double类型的最小值可以表示为 -DBL_MAX；DBL_MIN是最小正值(2.225 ×10⁻³⁰⁸)，大于0
代码3(正确版本)：

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