leetcode之Maximum Average Subarray I(643)

题目:

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

Python代码1:(未AC)

class Solution:
    def findMaxAverage(self, nums, k):
        maximum = -float('inf')
        for i in range(len(nums)-k+1):
            maximum = max(maximum,sum(nums[i:i+k]))
        return maximum/k

python代码2:

class Solution:
    def findMaxAverage(self, nums, k):
        maximum = -float('inf')
        res = 0
        for i in range(len(nums)):
            res += nums[i]
            if i >= k:
                res -= nums[i-k]
            if i >= k-1:
                maximum = max(maximum, res/k)
        return maximum

心得:代码1比较简洁,但是在测试比较长的案例时没有通过(考虑是由于sum的时间复杂度偏大),于是考虑滑动窗口机制,如以上代码2,成功解决问题。

猜你喜欢

转载自blog.csdn.net/cuicheng01/article/details/81391458