[Baekjoon] 2847 게임을 만든 동준이 python

🏷 문제

💡 코드

from sys import stdin

N = int(stdin.readline())
levels = [int(stdin.readline()) for _ in range(N)]

res = 0

# 뒤에서부터 비교
for i in range(N - 1, 0, -1):
    # 현재 값이 바로 앞의 값보다 작다면
    if levels[i] <= levels[i-1]:
        # 두 수의 차이 + 1 한 값을 결과값에 더함
        res += levels[i-1] - levels[i] + 1
        # 값 업데이트
        # ex) 7 5 였다면 res = 3, 7은 5-1 4로 업데이트
        levels[i-1] = levels[i] - 1
print(res)

🔑

입력받은 수가 연속으로 같은 수가 나올 수도 있고, 시간제한이 1초라는 점에서 리스트를 역순으로 조회해야겠다고 생각했다!