【力扣-每日小练】643. 子数组最大平均数 I（python)

643. 子数组最大平均数 I

题目链接：https://leetcode-cn.com/problems/maximum-average-subarray-i/
难度：简单
标签：数组（array）

题目描述

给定 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]。

代码样例

# 样例1：这是第一次尝试的结果，逻辑上应该没问题，就是在处理大量数据集时会导致用时超时
#		超时原因：每次滑动一格，就重新计算滑块内的和，计算量有很大的冗余
class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        # float('inf') 表示正无穷
        # -float('inf') 或 float('-inf') 表示负无穷
        average = float('-inf') # 负无穷
        temp = 0
        for i in range(len(nums) - k + 1):
            temp = sum(nums[i:i+k])
            if temp > average:
                average = temp

        return average/k

# 样例2：改进的方法，不至于每次滑动一次，就重新计算滑块内的和，
#		只需要加上新滑块内的最后一个，再减去上一滑块的第一个。
class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        temp=average = sum(nums[0:k])
        for i in range(1,len(nums) - k + 1):
            temp = temp + nums[i+k-1] - nums[i-1]
            if temp > average:
                average = temp

        return average/k