[백준/BOJ] gold5 - 14719번 빗물 (Python)

▶14719 - 빗물

백준 로고

▶문제

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

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

 

▶입력

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

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

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

 

▶출력

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

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

728x90

▶풀이

처음에는 그냥 모두 확인하는 방식으로 풀었다.

당연히 시간 초과가 날줄 알았는데, 그대로 통과가 됐다.

h, w = map(int, input().split())
arr = [[0 for _ in range(h)] for _ in range(w)]
block = list(map(int, input().split()))
answer = 0

for i in range(w):
    n = block[i]
    for j in range(n):
        arr[i][j] = 1

for j in range(h):
    check = 0
    cnt = 0
    for i in range(w):
        if arr[i][j] == 1:
            if check == 1:
                answer += cnt
                cnt = 0
            check = 1
        else:
            if check == 1:
                cnt += 1

print(answer)

 

그래도 조금 더 빠른 방법이 있지 않을까 하고 고민해 봤다.

물을 채워야 하는 곳 좌우로 가장 높은 블록을 구했다.

그만큼은 물이 차게 되기 때문이다.

 

그다음에, 둘 중에 min값을 가지고 현재 위치의 블록 개수를 빼고 물을 채워줬다.

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

for i in range(1, w - 1):
    left = max(block[:i])
    right = max(block[i + 1:])

    water = min(left, right)

    if water > block[i]:
        answer += (water - block[i])
    
print(answer)

메모리는 거의 동일한 메모리를 사용하고,

시간을 따져보면 거의 절반으로 줄었다.

코드의 길이도 절반 정도로 줄었다.

(중간 컴파일 에러는 ctrl+v 할 때, 다른 문자가 같이 입력돼서 에러가 발생한 거다.)