600-700/643.py

Author: 224apps

600-700/643.py

@@ -0,0 +1,28 @@
+'''
+Given an array consisting of n integers, find the contiguous subarray of given length k that has the maximum average value. And you need to output the maximum average value.
+
+Example 1:
+
+Input: [1,12,-5,-6,50,3], k = 4
+Output: 12.75
+Explanation: Maximum average is (12-5-6+50)/4 = 51/4 = 12.75
+ 
+
+Note:
+
+1 <= k <= n <= 30,000.
+Elements of the given array will be in the range [-10,000, 10,000].
+'''
+
+class Solution:
+    def findMaxAverage(self, nums: List[int], k: int) -> float:
+        tmp = 0
+        res = float('-inf')
+        for i, x in enumerate(nums):
+            tmp += x
+            if i >= k:
+                tmp -= nums[i-k]
+            if i >= k-1:
+                res = max(res, tmp)
+        res = res / k
+        return res