[백준/Python] 14719 빗물

문제 링크

14719번: 빗물

 

문제 풀이

• 양 끝을 제외하고 현재 위치에서 왼쪽, 오른쪽 부분에 자신보다 높이가 큰 블록이 있는지 확인하면 된다.
• 블록이 3 1 0 4 과 같을 때, 1 블록에서 왼쪽에는 3인 블록이 오른쪽에는 4인 블록이 있다. 그중 더 작은 블록만큼 물이 찰 수 있으므로 2만큼 차게 된다. 0 블록에서도 마찬가지이므로 3만큼 차게 된다.
• 만약, 한쪽이라도 자신의 위치보다 작은 블록이 있다면 물이 찰 수 없다.

 

전체 코드

import sys
input = sys.stdin.readline

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

res = 0
for i in range(1, w-1):
    # 현재 위치의 왼쪽, 오른쪽의 최대 블록 높이
    # 그중 더 작은 값
    height = min(max(block[:i]), max(block[i+1:]))
    # 현재 위치보다 큰 경우에만 물이 참
    if height >= block[i]:
        res += height - block[i]

print(res)