[백준] 14719번 빗물 - 파이썬(Python)

문제 (링크)

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

14719번: 빗물

 

나의 풀이

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)

 

코드 설명 및 풀이법

특정 위치에 물이 고이는지 고이지 않는지를 판단한 후 이를 코드로 잘 옮겨주기만 하면 된다.

 

특정 위치를 기준으로 양 옆에 자신보다 작은 높이의 블록이 있다면 해당 위치에는 물이 고일 수 없다.

 

다시 말해, 특정 위치에 물이 고이기 위해선 자신보다 더 높은 블록으로 왼쪽과 오른쪽이 둘러싸여 있어야한다.

 

그리고 이 조건을 만족할 때 물이 고이는 양은 왼쪽과 오른쪽 블록 중 더 낮은 블록과 현 위치의 값이다.

 

만약 높이가 1인 구역을 왼쪽 오른쪽으로 각각 3, 4 높이의 블록이 감싸고 있다면 해당 구역은 더 낮은 블록인 3에서 현 위치의 높이인 1을 뺀 2 만큼의 물이 고인다.

 

첫 번째 칸과 마지막칸은 물이 고일 수 없으므로 range를 1부터 w - 1까지 잡아주고 각 위치마다 알맞는 양의 물을 채워넣으면 답을 구할 수 있다.