mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-01 18:55:23 +08:00
Yudong Jin
2f561d3d51
* feat(ko): add the complete Korean online edition * Use UTF-8 for Python CI output on Windows * Remove print-book promotion from the Korean site * fix(ko): preserve emphasis before Korean particles
34 lines
850 B
Python
34 lines
850 B
Python
"""
|
|
File: max_capacity.py
|
|
Created Time: 2023-07-21
|
|
Author: krahets ([email protected])
|
|
"""
|
|
|
|
|
|
def max_capacity(ht: list[int]) -> int:
|
|
"""최대 용량: 탐욕"""
|
|
# i와 j가 각각 배열의 양 끝을 가리키도록 초기화한다
|
|
i, j = 0, len(ht) - 1
|
|
# 최대 용량의 초깃값은 0이다
|
|
res = 0
|
|
# 두 칸막이가 만날 때까지 탐욕적 선택을 반복한다
|
|
while i < j:
|
|
# 최대 용량을 갱신한다
|
|
cap = min(ht[i], ht[j]) * (j - i)
|
|
res = max(res, cap)
|
|
# 낮은 칸막이를 안쪽으로 이동한다
|
|
if ht[i] < ht[j]:
|
|
i += 1
|
|
else:
|
|
j -= 1
|
|
return res
|
|
|
|
|
|
"""실행 코드"""
|
|
if __name__ == "__main__":
|
|
ht = [3, 8, 5, 2, 7, 7, 3, 4]
|
|
|
|
# 탐욕 알고리즘
|
|
res = max_capacity(ht)
|
|
print(f"최대 용량은 {res}이다")
|