[백준 14719 🥇] 빗물

문제

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

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

입력

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

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

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

출력

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

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

문제풀이

import sys

h,w = map(int,sys.stdin.readline().split(' '))
blocks=list(map(int, sys.stdin.readline().split(' ')))

total=0
for i in range(1,w-1):
    left_max=max(blocks[:i])
    right_max=max(blocks[i+1:])
    add_value=min(left_max, right_max)-blocks[i]
    if add_value>0:
        total+=add_value

print(total)

아이디어만 떠올리면 간단한 문제다!

1번부터 w-1번 인덱스까지 탐색하면서 현재 위치에서 왼쪽 블록들을 기준으로 가장 큰 값을 left_max에 저장하고, 오른쪽 블록들을 기준으로 가장 큰 값을 right_max에 저장한다. 

그리고 left_max와 right_max 중 더 작은 값에서 현재 위치의 블록 크기를 빼서 add_value에 저장하고, 만약 add_value가 0보다 크다면 고이는 물이 있다는 뜻이므로 고이는 빗물의 총량(total)에 더한다.

 

이 풀이는 그림으로 이해하면 더 쉽다.

문제에서 제공한 예제2번의 경우를 그림으로 표현하면 다음과 같다.