서론

그리디 문제입니다.
현재 사과 위치와 바구니의 시작과 끝의 위치를 계산하여 바구니 이동을 구현하면 되는 문제입니다.

난이도

실버 5

---

문제

조건

시간 제한	메모리 제한
1 초	128 MB

---

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 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

---

풀이

1. 사과의 위치를 입력받는다.
2. 이동할 거리를 저장할 변수(tmp)를 생성한다.
3. 사과의 위치에 따라 바구니의 위치(시작 지점)를 이동시킨다.
   (바구니의 크기는 m으로 고정이기 때문에 시작 지점만 옮겨도 끝 지점은 그에 맞춰진다.)
4. 바구니가 이동한 만큼 총 이동 거리에 더해준다.

코드

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
s = 1		# 처음 시작 위치
ans = 0		# 총 이동 거리

for _ in range(int(input())):
    apple = int(input())
    tmp = 0	# 이번 차례에 이동할 거리 (사과가 바구니 범위 안에 있다면 움직이지 않음)

    if apple < s:					# 사과가 바구니의 시작 지점보다 앞에 있는 경우
        tmp = s - apple				# 이동할 거리는 시작지점 - 사과의 위치
        s -= tmp					# 바구니 왼쪽으로 이동
    elif apple > s + m - 1:			# 사과가 바구니의 끝 지점보다 뒤에 있는 경우
        tmp = apple - (s + m - 1)	# 이동할 거리는 사과의 위치 - 끝지점
        s += tmp					# 바구니 오른쪽으로 이동
    
    ans += tmp						# 총 이동거리 +

print(ans)

실행 결과