[백준] 2828번: 사과 담기 게임/ 파이썬

문제 링크

---

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

2828번: 사과 담기 게임

 

풀이 코드

---

n, m = map(int, input().split())
j = int(input())

start = 1
end = m
distance = 0

for _ in range(j):
    p = int(input())

    if p < start:
        distance += (start - p)
        start = p
        end = p + m - 1

    elif p > end:
        distance += (p - end)
        end = p
        start = end - m + 1

print(distance)

코드 해석

---

수를 받은 뒤 바구니의 길이를 start와 end로 지정해놓는다.

사과가 떨어지는 위치와 바구니의 위치를 비교하여

distance와 start값과 end값을 조정한다.