문제
2차원 세계에 블록이 쌓여있다. 비가 오면 블록 사이에 빗물이 고인다.
비는 충분히 많이 온다. 고이는 빗물의 총량은 얼마일까?
입력
첫 번째 줄에는 2차원 세계의 세로 길이 H과 2차원 세계의 가로 길이 W가 주어진다. (1 ≤ H, W ≤ 500)
두 번째 줄에는 블록이 쌓인 높이를 의미하는 0이상 H이하의 정수가 2차원 세계의 맨 왼쪽 위치부터 차례대로 W개 주어진다.
따라서 블록 내부의 빈 공간이 생길 수 없다. 또 2차원 세계의 바닥은 항상 막혀있다고 가정하여도 좋다.
출력
2차원 세계에서는 한 칸의 용량은 1이다. 고이는 빗물의 총량을 출력하여라.
빗물이 전혀 고이지 않을 경우 0을 출력하여라.
풀이 코드
H, W = map(int, input().split())
rain = list(map(int, input().split()))
water = 0
for i in range(1, W-1): #빗물양 계산 , i는 가운데 기준
left = max(rain[:i]) #왼쪽에서 가장 높은 블록 높이
right = max(rain[i+1:]) #오른쪽에서 가장 높은 블록 높이
# 현재 열에서 고이는 빗물의 양은
# 현재 열의 높이보다 왼쪽과 오른쪽에서 가장 높은 블록의 높이 중 작은 값에서
# 현재 열의 높이를 뺀 값
water += max(0, min(left, right) - rain[i])
print(water)
완전 탐색으로 품