코테/BOJ

백준 14719 [빗물] 파이썬

동 캄 2022. 9. 3. 15:59
반응형

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

 

14719번: 빗물

첫 번째 줄에는 2차원 세계의 세로 길이 H과 2차원 세계의 가로 길이 W가 주어진다. (1 ≤ H, W ≤ 500) 두 번째 줄에는 블록이 쌓인 높이를 의미하는 0이상 H이하의 정수가 2차원 세계의 맨 왼쪽 위치

www.acmicpc.net

# 백준 14719 빗물
H, W = map(int, input().split())
lst = list(map(int, input().split()))
temp = [0]*(H+1)
i = 0
ans = 0
M = lst[0]
while i < W:
    j = 1
    while i+j < W:
        if lst[i+j] <= lst[i+j-1]:
            for k in range(lst[i+j], M):
                temp[k] += 1
        else:
            if lst[i+j] >= M:
                for k in range(0, M):
                    ans += temp[k]
                    temp[k] = 0
                M = lst[i+j]
                break
            else:
                for k in range(0, lst[i+j]):
                    ans += temp[k]
                    temp[k] = 0
                for k in range(lst[i+j], M):
                    temp[k] += 1
        j += 1
    i += j
print(ans)

웅덩이에 저장될 수 있는 물의 양 temp 배열을 생성하고 여기에 값을 저장한다.

 

반응형