[BOJ] 2828. 사과 담기 게임 (python)

✏️ 문제

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다. (M<N) 플레이어는 게임을 하는 중에 바구니를 왼쪽이나 오른쪽으로 이동할 수 있다. 하지만, 바구니는 스크린의 경계를 넘어가면 안 된다. 가장 처음에 바구니는 왼쪽 M칸을 차지하고 있다.
스크린의 위에서 사과 여러 개가 떨어진다. 각 사과는 N칸중 한 칸의 상단에서 떨어지기 시작하며, 스크린의 바닥에 닿을때까지 직선으로 떨어진다. 한 사과가 바닥에 닿는 즉시, 다른 사과가 떨어지기 시작한다.
바구니가 사과가 떨어지는 칸을 차지하고 있다면, 바구니는 그 사과가 바닥에 닿을 때, 사과를 담을 수 있다. 상근이는 사과를 모두 담으려고 한다. 이때, 바구니의 이동 거리의 최솟값을 구하는 프로그램을 작성하시오.

 

🤖 알고리즘

 

#구현  #그리디

 

 

🤯 풀이 방법

사과의 개수가 최대 20개라 범위가 작은 문제여서 부담없이 풀 수 있었다.

가장 왼쪽에서 시작한다고 했으니 1~M으로 현재 상자의 위치를 담고,

들어오는 사과의 위치를 파악하며 현재 범위에서 받을 수 없는 위치에 있으면 최소한으로 이동하고 상자 위치를 조정한다

 

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
j = int(input())  # 사과의 개수

start = 1
end = M
ans = 0

for i in range(j):
    cur = int(input())
    if cur < start:
        g = abs(cur - start)
        ans += g
        start = cur
        end -= g
        
    elif cur > end:
        g = abs(cur - end)
        ans += g
        end = cur
        start += g

print(ans)

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

2828번: 사과 담기 게임