-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcontainer_with_most_water.py
33 lines (25 loc) · 1.29 KB
/
container_with_most_water.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from typing import List
def test_1():
assert Solution().maxArea([1, 8, 6, 2, 5, 4, 8, 3, 7]) == 49
def test_2():
assert Solution().maxArea([1, 1]) == 1
class Solution:
def maxArea(self, height: List[int]) -> int:
left_ptr = 0
right_ptr = len(height) - 1
max_amount = 0
# Идем с лева на право и с права на лево
while left_ptr < right_ptr:
# Ищем площадь, min потому что вода выливается через наименьшую стенку
amount = min(height[left_ptr], height[right_ptr]) * (right_ptr - left_ptr)
# Если она больше текущей, то устанавливаем ее в качестве текущей
if amount > max_amount:
max_amount = amount
# Двигаем указатель в зависимости от того, какая стенка наименьшая.
# Левый, если левая, в противном случае правый.
# Задача - максимизировать площадь.
if height[left_ptr] <= height[right_ptr]:
left_ptr += 1
else:
right_ptr -= 1
return max_amount