feat: [LeetCode #1732] Find The Highest Altitude (#5)

Тип: prefix sum
Сложность: easy
Временная сложность: O(n)
Пространственная сложность: O(1)

-

Ссылка: https://leetcode.com/problems/find-the-highest-altitude/

Author: onixlas

src/prefix_sum/__init__.py

@@ -0,0 +1,13 @@
+from typing import List
+
+
+class Solution:
+    def largestAltitude(self, gain: List[int]) -> int:
+        current_altitude = 0
+        max_altitude = 0
+
+        for change in gain:
+            current_altitude += change
+            max_altitude = max(max_altitude, current_altitude)
+
+        return max_altitude

src/prefix_sum/find_the_highest_altitude/__init__.py

@@ -0,0 +1,14 @@
+import pytest
+from src.prefix_sum.find_the_highest_altitude.solution import Solution
+
+
+@pytest.mark.parametrize(
+    "gain, expected",
+    [
+        ([-5, 1, 5, 0, -7], 1),
+        ([-4, -3, -2, -1, 4, 3, 2], 0),
+    ],
+)
+def test_largest_altitude(gain, expected):
+    solution = Solution()
+    assert solution.largestAltitude(gain) == expected