[BOJ] 14719번 빗물 파이썬

문제

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

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

입력

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

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

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

출력

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

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

 

풀이

#시뮬레이션

양쪽으로 블록으로 막혀있는 만큼만 물을 채울 수 있다.

따라서 블록이 양쪽으로 막혀있는 높이를 구해야 한다. left, right 배열을 만든 후 입력받은 배열을 깊은 복사를 진행한 후 각 인덱스 위치에서 왼쪽기준으로 가장 높은 블록의 높이, 오른쪽 기준으로 가장 높은 블록의 높이를 구한다.

이어서 인덱스 위치에 물을 담을 수 있는 양은 right와 left 중 작은 값에서 해당 인덱스 위치의 블록 높이 만큼 빼주는 방법으로 구할 수 있다.

코드

import copy

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

arr = list(map(int, input().split()))
left = copy.deepcopy(arr)
right = copy.deepcopy(arr)

for i in range(1, len(arr)):
    left[i]=max(left[i], left[i-1])

for i in range(len(right)-2, -1, -1):
    right[i]=max(right[i], right[i+1])

sum=0
for i in range(W):
    sum+=(min(right[i], left[i])-arr[i])

print(sum)