[백준][Python] 14719 빗물

구현, 시뮬레이션 / 골드 5

☔️

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

14719번: 빗물

 

 

내 풀이

H, W = map(int, input().split())
arr = [[0] * W for _ in range(H)]
height = list(map(int, input().split()))
ans = 0

for i in range(W):
    for j in range(H-height[i], H):
        arr[j][i] = 1

for i in range(H):
    block = []
    for j in range(W):
        if arr[i][j] == 1:
            block.append(j)
    if len(block) > 1:
        s = block.pop(0)
        e = block.pop(0)
        ans += e - s - 1

        while block:
            s = e
            e = block.pop(0)
            ans += e - s - 1

print(ans)

 

 

코드 설명

 

높이를 입력받아 각 열에 해당 높이만큼 블록을 쌓는다.

값이 1인 것을 블록으로 간주한다.

 

for i in range(W):
    for j in range(H-height[i], H):
        arr[j][i] = 1

 

이제 한 행씩 검사한다.

 

block: 현재 행에서 블록의 인덱스를 저장하는 리스트

 

빗물이 고이려면 블록이 무조건 2개 이상이어야 한다.

첫 번째 블록의 위치를 s(시작 블록)에 저장하고, 두 번째 블록의 위치를 e(끝 블록)에 저장한 후,

두 블록 사이에 고인 빗물량을 ans에 더한다.

 

모든 블록을 검사할 때까지 이 과정을 반복한다.

다음 블록으로 넘어갈 때, 끝 블록이 시작 블록(s = e)이 된다.

for i in range(H):
    block = []
    for j in range(W):
        if arr[i][j] == 1:
            block.append(j)
    if len(block) > 1:
        s = block.pop(0)
        e = block.pop(0)
        ans += e - s - 1

        while block:
            s = e
            e = block.pop(0)
            ans += e - s - 1