알고리즘 문제/Leetcode

643. Maximum Average Subarray I

BEstyle 2023. 4. 2. 19:36

You are given an integer array nums consisting of n elements, and an integer k.

Find a contiguous subarray whose length is equal to k that has the maximum average value and return this value. Any answer with a calculation error less than 10-5 will be accepted.

 

Example 1:

Input: nums = [1,12,-5,-6,50,3], k = 4
Output: 12.75000
Explanation: Maximum average is (12 - 5 - 6 + 50) / 4 = 51 / 4 = 12.75

Example 2:

Input: nums = [5], k = 1
Output: 5.00000

 

Constraints:

  • n == nums.length
  • 1 <= k <= n <= 105
  • -104 <= nums[i] <= 104

class Solution:
    def findMaxAverage(self, nums: List[int], k: int) -> float:
        count=0
        left=0
        right=k-1
        for i in range(k):
            count += nums[i]
        ans=count
        for i in range(len(nums)-k):
            count-=nums[left]
            left+=1
            right+=1
            count+=nums[right]
            ans=max(ans,count)
        return ans/k