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

题目

给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数。

示例
输入: [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]。

思路

例如 , 输入 [0,1,2,3,4,5,6] , k = 4
有三个关键变量sum,max,x
sum 初始化为最开始的k个数,sum = 0+1+2+3,以后每次迭代都是减去第一个数,加上后面一个数
max用于记录最大的和
x 用于计算本次的k个数的和
代码如下:

class Solution:
    def findMaxAverage(self, nums,k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: float
        """
        lens = len(nums)
        sum = 0
        max = 0
        for i in range(k):  #初始化
            sum = sum + nums[i]
            max = sum
        for i in range(1,lens-k+1):
            x = sum + nums[k+i-1] - nums[i-1]           
            if x>max:
                max = x
            sum = x
        return max/k

学习他人代码

class Solution:
    def findMaxAverage(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: float
        """
        cur = sum(nums[:k]) #初始化,计算前k项
        max_val = cur
        for i in range(1, len(nums)-k+1):
            cur += nums[i+k-1]-nums[i-1]
            if cur > max_val:
                max_val = cur
        return max_val/k

这段代码的思路和我的一样,但是比我快,主要是以下几点:

  1. 初始化计算和的时候,我用的for循环,而大佬用的sum()函数
  2. 在我的代码中x其实是多余的

你可能感兴趣的:(python)