[Python | 백준 2828번] 사과 담기 게임

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

 

문제

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다. (M<N) 플레이어는 게임을 하는 중에 바구니를 왼쪽이나 오른쪽으로 이동할 수 있다. 하지만, 바구니는 스크린의 경계를 넘어가면 안 된다. 가장 처음에 바구니는 왼쪽 M칸을 차지하고 있다.

스크린의 위에서 사과 여러 개가 떨어진다. 각 사과는 N칸중 한 칸의 상단에서 떨어지기 시작하며, 스크린의 바닥에 닿을때까지 직선으로 떨어진다. 한 사과가 바닥에 닿는 즉시, 다른 사과가 떨어지기 시작한다.

바구니가 사과가 떨어지는 칸을 차지하고 있다면, 바구니는 그 사과가 바닥에 닿을 때, 사과를 담을 수 있다. 상근이는 사과를 모두 담으려고 한다. 이때, 바구니의 이동 거리의 최솟값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 M이 주어진다. (1 ≤ M < N ≤ 10) 둘째 줄에 떨어지는 사과의 개수 J가 주어진다. (1 ≤ J ≤ 20) 다음 J개 줄에는 사과가 떨어지는 위치가 순서대로 주어진다.

출력

모든 사과를 담기 위해서 바구니가 이동해야 하는 거리의 최솟값을 출력한다.

 

---

내 풀이

바구니 길이에 집중해서 사과 위치를 받을 수 없는지 있는지를 체크했다. 바구니의 길이가 2라면 처음 위치는 1에서부터 2에 있으니 [1, 2]로 두고사과의 위치가 3이라면 가장 끝 길이에서 최소로 움직이기 위해 3(사과위치) - 2(바구니 최대 위치) = 1 에 맞추어 이동시켰다. 반대로 [4, 5] 위치에 있는데 사과가 1에서 떨어진다면 1(사과위치) - 4(바구니 최소 위치) = -3 만큼 움직이는 것이다. 움직이면 바구니 위치도 갱신해주고 [4, 5] -> [1, 2] 최종 움직인 거리도 절댓값으로 더해주는 식으로 구했다.  

 

# 31120kb 32ms

n, m = map(int, input().split())
tot_apple = int(input())
basket = [1, m]
total_move = 0

for i in range(tot_apple):
  pos = int(input())
  if basket[0] <= pos <= basket[1]:
    move = 0
  elif pos < basket[0]:
    move = (pos - basket[0])
  elif pos > basket[1]:
    move = (pos - basket[1])
  basket[0] += move
  basket[1] += move
  total_move += abs(move)

print(total_move)

 

메모

그리디 알고리즘, 실버 5