[BOJ:2828] 사과 담기 게임

문제 링크 : 사과 담기 게임

[2828번: 사과 담기 게임

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다. (M<n) 플레이어는="" 게임을="" 하는="" 중에="" 바구니를="" <="" p=""> </n)>

www.acmicpc.net](https://www.acmicpc.net/problem/2828)

문제

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

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

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

입력

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

출력

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

예제 입력 1 복사

5 1
3
1
5
3

예제 출력 1 복사

6

예제 입력 2 복사

5 2
3
1
5
3

예제 출력 2 복사

4

해설

구역이 N 칸으로 나뉘어진 공간이 있고, M칸을 차지하는 바구니로 좌, 우 이동을 한다. 그리고 바구니가 사과가 떨어지는 칸을 차지하고 있다면, 사과를 담을 수 있는데, 이대 이 구역에서 떨어지는 사과를 모두 담는 이동거리의 최솟값을 구하는 문제이다.

바구니가 시작 위치가 가장 왼쪽 부터 시작한다고 하였으므로 이 장소를 1로 정의하고 사이즈-1 만큼 바구니의 가장 우측 차지 칸을 표시하면, 숫자로 바구니의 위치를 표시할 수있다. 여기서 바구니 왼쪽 차지칸에서 사이즈만큼이 아니라 사이즈-1 만큼 더한 값을 바구니의 우측 차지 칸을 표시하는 이유는 더하기는 현재 위치를 제외하고 그 다음 칸부터 더해진 것을 표시하기 때문이다. 예를 들면 사이즈가 1인 바구니의 왼쪽 위치는 1이고 우측 위치는 1이다. 

이렇게 숫자로 바구니의 위치를 표시해주고 사과의 위치가 이 위치 사이에 표시 되어 있다면 이동하지 않고 아니라면 이동을 하면 된다. 이때 이동하는 방향은 좌, 우 두가지로 나뉠 수 있다. 좌로 이동하는 경우는 사과의 위치가 바구니의 왼쪽에 있을 때(사과 위치 값이 바구니의 위치보다 작으면) 이고, 우로 이동하는 경우는 사과의 위치가 바구니의 우측에 있을때(사과 위치 값이 바구니의 위치보다 크면) 이다.

그리고 좌로 이동하는 경우라면 사과의 위치와 바구니 좌측 위치와의 차이, 우로 이동하는 경우라면 사과의 위치와 바구니의 우측 위치 값이 만큼만 딱 이동하면 최소 이동 경로를 구할 수 있다.

코드

import sys

input = sys.stdin.readline
leftmost = 1

screen, basket_size = map(int,input().split())

number_of_apples = int(input())
basket_left = leftmost
basket_right = leftmost + basket_size - 1
move = 0

for _ in range(number_of_apples):
    apple_position = int(input())

    if basket_left <= apple_position <= basket_right: 
        continue
    else: 
        if apple_position < basket_left:
            cur_move = basket_left - apple_position
            basket_left = basket_left - cur_move
            basket_right = basket_right - cur_move
            move += cur_move
        elif apple_position > basket_right:
            cur_move = abs(apple_position - basket_right)
            basket_left = basket_left + cur_move
            basket_right = basket_right + cur_move 
            move += cur_move
print(move)