[백준] #14719 빗물 파이썬

https://www.acmicpc.net/problem/14719

 

문제

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

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

입력

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

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

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

출력

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

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

---

복기

1. H x W의 array를 만들어서 완전탐색을 해야하나 싶었는데, 열 단위로 탐색을 하면 되는 문제였다

2. 문제를 단순화시켜서 생각하는 습관을 들여야겠다

솔루션

H, W = map(int, input().split())
world = list(map(int, input().split()))

cnt = 0

for i in range(1, W-1):
    left_max = max(world[:i])
    right_max = max(world[i+1:])
    min_height = min(left_max, right_max)
    if world[i] < min_height:
        cnt += min_height - world[i]

print(cnt)