Merge pull request #2 from onixlas/feature/max-average-subarray-i

feat: add maximum average subarray i

Author: onixlas

src/sliding_window/__init__.py

@@ -0,0 +1,14 @@
+from typing import List
+
+
+class Solution:
+    def findMaxAverage(self, nums: List[int], k: int) -> float:
+        total = sum(nums[:k])
+        max_total = total
+
+        for index in range(k, len(nums)):
+            total -= nums[index - k]
+            total += nums[index]
+            max_total = max(total, max_total)
+
+        return max_total / k

src/sliding_window/maximum_average_subarray/__init__.py

@@ -0,0 +1,11 @@
+import pytest
+from src.sliding_window.maximum_average_subarray.solution import Solution
+
+
+@pytest.mark.parametrize(
+    "nums, k, expected",
+    [([1, 12, -5, -6, 50, 3], 4, 12.75), ([5], 1, 5)],
+)
+def test_max_average_subarray(nums, k, expected):
+    solution = Solution()
+    assert solution.findMaxAverage(nums, k) == expected