LeetCode(N0.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]。

方法1:直接遍历,计算超时

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        n = len(nums)
        if n==0 or n<k or k==0:
            return 0
        else:
            maxN = sum(nums[0:k])/k
            for i in range(n-k+1):
                if maxN < sum(nums[i:i+k])/k:
                    maxN = sum(nums[i:i+k])/k
                
            return maxN

方法2:在1的基础上优化,下一轮迭代只需要用上一次的结果减去上一轮数组的第一个值,然后加上接下来的那个值

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        n = len(nums)
        if n==0 or n<k or k==0:
            return 0
        else:
            maxN = sum(nums[0:k])
            maxRes = maxN
            for i in range(n-k):
                maxN = maxN - nums[i] + nums[i+k]
                if maxN > maxRes:
                    maxRes = maxN
                
            return maxRes/k

猜你喜欢

转载自blog.csdn.net/zuolixiangfisher/article/details/89062144
今日推荐