[Python] 백준 14719번 - 빗물 (골드 5)

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 시뮬레이션
 - 구현

 

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

14719번: 빗물

 

문제

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

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

입력

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

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

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

출력

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

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

---

조금 생각하면 풀 수 있는 문제였다.

 

현재 위치에서 왼쪽의 최대기둥, 오른쪽 최대기둥을 찾아서 현재 위치에 고일 수 있는 물의 개수를 구하면 된다.

 

import sys
input = sys.stdin.readline

h, w = map(int, input().split())
block = list(map(int, input().split()))

def find_left(now):
    height = block[now]
    for i in range(now, -1, -1):
        if block[i] >= height:
            height = block[i]
    return height

def find_right(now):
    height = block[now]
    for i in range(now, w):
        if block[i] >= height:
            height = block[i]
    return height

cnt = 0
for now in range(w):
    left = find_left(now)
    right = find_right(now)
    cnt += min(left, right) - block[now]
    
print(cnt)