프로그래밍/Algorithm

백준 2828 사과 담기 게임 파이썬

모지사바하 2021. 3. 8. 16:35
N, M = map(int, input().split())
apple_cnt = int(input())
A = []

for i in range(apple_cnt):
    A.append(int(input()))
    
ans = 0
box_pos = 1
dif = 0
for i in range(len(A)):
    apple_pos = A[i]
    if apple_pos < box_pos:
        dif= abs(box_pos - apple_pos)
        box_pos-= dif
        ans+=dif
    elif box_pos + M - 1 < apple_pos:
        dif= abs(box_pos + M - 1 - apple_pos)
        box_pos+=dif
        ans+=dif
    
    
print(ans)

엄청 쉬운 문제인데,

정신이 산만해서 그런가.. 집중이 잘안돼서 약간의 시간이 소비된 문제

이런문제는 그자리에서 바로 답이 나와야하는거 아닌가..

 

그리디 최고 순한 맛.