[Baekjoon] 백준 14719 '빗물' 문제풀이 Python, 파이썬, 알고리즘 정리

📝 14719 문제 

2차원 세계에 블록이 쌓여있다. 비가 오면 블록 사이에 빗물이 고인다.

비는 충분히 많이 온다. 고이는 빗물의 총량은 얼마일까?

 

 

입력 

첫 번째 줄에는 2차원 세계의 세로 길이 H과 2차원 세계의 가로 길이 W가 주어진다. (1 ≤ H, W ≤ 500)

두 번째 줄에는 블록이 쌓인 높이를 의미하는 0이상 H이하의 정수가 2차원 세계의 맨 왼쪽 위치부터 차례대로 W개 주어진다.

따라서 블록 내부의 빈 공간이 생길 수 없다. 또 2차원 세계의 바닥은 항상 막혀있다고 가정하여도 좋다.

 

출력

2차원 세계에서는 한 칸의 용량은 1이다. 고이는 빗물의 총량을 출력하여라.

빗물이 전혀 고이지 않을 경우 0을 출력하여라.

 

🧐 풀이

첫 번째 칸과 마지막 칸에는 빗물이 고이지 않는다. 따라서 반복문의 범위를 1부터 W-1까지로 잡아줘야 한다.
빗물이 고이기 위해서는 양쪽(left, right)으로 자신보다 높은 블록이 있어야 한다.
left와 right 중 더 낮은 블록과 자신을 뺀 값이 물이 고이는 양이다.

 

H, W = map(int, input().split())
num = list(map(int, input().split()))
res = 0
for i in range(1, W-1) :
    left = max(num[:i])
    right = max(num[i+1:])
    tmp = min(left, right)
    if num[i] < tmp : 
        res += tmp-num[i]
        
print(res)