[BOJ][🟡5][백준#14719] 빗물

문제 출처

BAEKJOON Online Judge #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

출력

5

예제 2

입력

4 8
3 1 2 3 4 1 1 2

출력

5

예제 3

입력

3 5
0 0 0 2 0

출력

0

My Sol

import sys
input = sys.stdin.readline

H, W = map(int, input().split())
heights = list(map(int, input().split()))

# 최대 높이 계산
max_h = max(heights)

# 최대 넓이 계산
# 앞에서부터 최대 높이까지 누적
ssum = 0
si = 0
cur_h = 0
while heights[si] < max_h:
    cur_h = max(cur_h, heights[si])
    ssum += cur_h
    si += 1

# 뒤에서부터 최대 높이까지 누적
ei = W-1
cur_h = 0
while heights[ei] < max_h:
    cur_h = max(cur_h, heights[ei])
    ssum += cur_h
    ei -= 1

# 최대높이구간 누적
ssum += (ei-si+1)*max_h

print(ssum - sum(heights))

전체 공간을 3등분하여 비가 고였을 경우의 면적을 계산하고, 이를 실제 블록의 면적으로 빼면 비의 면적만 구할 수 있다.

블록의 최대 높이를 구하고, 앞에서부터 블록의 최대높이에 다다를 때까지 현재 높이와 이전 최대 높이를 비교하여 더 큰 값을 계속 누적하며 index를 높여간다. 뒤에서도 마찬가지이다.

그렇다면 두 index 사이의 공간에 최대높이가 있는 것이고, 양쪽에서 최대 높이에 도달했다면, 그 사이의 공간은 모두 최대높이와 인덱스 차의 곱으로 보아야 하기 때문에 나중에 더해준다.

결과

맞았습니다!!

모범답안

출처

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

ans = 0
for i in range(1, w - 1):
    left_max = max(world[:i])
    right_max = max(world[i+1:])

    compare = min(left_max, right_max)

    if world[i] < compare:
        ans += compare - world[i]

print(ans)

풀이의 길이가 짧은 코드를 발견해서 분석해보려 한다. 각 자리마다 해당 지점을 기점으로 좌우의 최대 높이 중 낮은 높이를 계산하고, 현재 지점의 높이가 좌우 최대 높이보다 낮다면 좌우 최대 높이와 현재 높이의 차만큼 비가 고이는 것이므로 ans에 더해준다.

이렇게도 풀 수 있겠으나 W가 커지면 매번 max를 구하는 로직 자체가 오래 걸리고, 이러한 반복이 더 많이 반복되어야 하므로 연산시간이 몹시 높아질 것이다.

좋은 풀이라고 보기에는 아쉽고 짧게 잘 풀어낸 코드인 것 같다.