[백준] 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:

 

문제 풀이

이문제는 한칸씩 물이 얼마나 찰지 계산해서 더하는 방식으로 접근한다.

특정 위치 i 에 빗물이 고이려면, 내 왼쪽에서 가장 높은 벽과 내 오른쪽에서 가장 높은 벽 사이에 갇혀야한다. 

1. 반복 : 첫번쨰와 마지막 블록을 제외한 모든 칸을 하나씩 확인한다.

2. 탐색: 현재 칸 i를 기준으로 

   - 왼쪽 전체에서 가장 높은벽을 찾는다 

   - 오른쪽 전체에서 가장 높은 벽을 찾는다

3. 계산 : 두 벽 중 더 낮은 벽을 기준으로 물이 채워진다.

   - 기준 높이 = min(left_max,right_max)

   - 만약 기준 높이가 현재 내 높이보다 높으면, 그 차이만큼 물이 고이낟. 

 

import sys

H,W=map(int, sys.stdin.readline().split())
blocks=list(map(int,sys.stdin.readline().split()))

answer = 0

# 맨 왼쪽(0)과 맨 오른쪽 (W-1) 에는 물이 고일 수 없으므로 1부터 W-2 까지 반복
for i in range(1,W-1):
    # 현재 위치를 기준으로 왼쪽에서 가장 높은 벽과 오른쪽에서 가장 높은 벽 찾기 
    left_max=max(blocks[:i])
    right_max=max(blocks[i+1:])
    
    # 양쪽 벽 중 더 낮은 벽까지 
    compare=min(left_max,right_max)
    
    # 기준 벽이 현재 블록보다 높아야 물이 고인다 
    if compare>blocks[i]:
        answer+=compare-blocks[i]
        
print(answer)

 

- blocks[:i] 내 위치보다 왼쪽에 있는 모든 블록들이다

- blocks[i+1:] 내 위치보다 오른쪽에 있는 모든 블록들이다

- min(left_max,right_max): 물은 양쪽 벽 중 낮은 쪽까지만 찰 수 있다.