[백준]14719번: 빗물

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

14719번: 빗물

---

h*w 크기의 이차원 배열을 만들어서 블록이 있는 칸은 1, 빈 칸은 0으로 표기하였다.

 

0인 칸 중에서, 같은 높이의 왼쪽편에 블록이 1개 이상 존재하고 오른편에도 블록이 1개 이상 존재하면 빗물이 고인다.

w는 최대 500으로 매우 작은 수이기 때문에 그냥 리스트 슬라이싱으로 확인해주었다.

 

python

import sys
input = sys.stdin.readline

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

block = [[0 for j in range(w)] for i in range(h)]
for i in range(w):
  for j in range(height[i]):
    block[j][i] = 1

answer = 0
for i in range(h):
  for j in range(w):
    if block[i][j] == 1:
      continue
    if 1 in block[i][:j] and 1 in block[i][j+1:]:
      answer += 1

print(answer)