[Baekjoon] 14719 빗물 python

🏷 문제

💡 코드

h, w = map(int,input().split())
blocks = list(map(int, input().split()))
h_max = 0

# 가장 큰 높이를 가진 블럭을 구함
for i in range(len(blocks)):
    if h_max < blocks[i]:
        h_max = blocks[i]
        h_idx = i

# 오른쪽 구역
res = 0
val = 0
for i in range(h_idx+1):
    if val < blocks[i]:
        val = blocks[i]
    res += val

# 왼쪽 구역
val = 0
for i in range(w-1, h_idx, -1):
    if val < blocks[i]:
        val = blocks[i]
    res += val
    
print(res-sum(blocks))

🔑