[🥇5 / 백준 14719 / 파이썬] 빗물

 

14719번: 빗물

 

문제

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

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

입력

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

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

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

출력

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

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

 

예제 입력 1 4 4 3 0 1 4	예제 출력 1 5
예제 입력 2 4 8 3 1 2 3 4 1 1 2	예제 출력 2 5
예제 입력 3 3 5 0 0 0 2 0	예제 출력 3 0

 

 

힌트

힌트 1:

힌트 2:

힌트 3:

 

풀이

 

가로줄 한 줄씩 나누어 생각한다.

공중에 떠 있는 벽은 없으므로, 벽과 벽 사이에 공간이 존재한다면 무조건 빗물이 고일 수 있다.

기존에 벽이 1번이라도 등장한 다음 빈 공간이 있다면 카운트한다.

그리고 다음 벽이 등장할 경우 카운트했던 것을 전체 답에 추가해주도록 한다.

 

import sys
input = sys.stdin.readline

n,m = map(int,input().split())
arr = list(map(int,input().split()))
board = [[0]*m for _ in range(n)]
for i in range(m):
    for j in range(arr[i]):
        board[j][i] = 1

ans = 0
for i in range(n):
    start = False
    s = 0
    for j in range(m):
        if board[i][j] == 1:
            if start == True and s>0:
                ans += s
                s=0
            start = True
        elif start == True:
            s += 1
print(ans)