https://www.acmicpc.net/problem/2828
상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다.
(M<N) 플레이어는 게임을 하는 중에 바구니를 왼쪽이나 오른쪽으로 이동할 수 있다. 하지만, 바구니는 스크린의 경계를 넘어가면 안 된다. 가장 처음에 바구니는 왼쪽 M칸을 차지하고 있다.
스크린의 위에서 사과 여러 개가 떨어진다. 각 사과는 N칸중 한 칸의 상단에서 떨어지기 시작하며, 스크린의 바닥에 닿을때까지 직선으로 떨어진다. 한 사과가 바닥에 닿는 즉시, 다른 사과가 떨어지기 시작한다.
바구니가 사과가 떨어지는 칸을 차지하고 있다면, 바구니는 그 사과가 바닥에 닿을 때, 사과를 담을 수 있다. 상근이는 사과를 모두 담으려고 한다. 이때, 바구니의 이동 거리의 최솟값을 구하는 프로그램을 작성하시오.
첫째 줄에 N과 M이 주어진다.
(1 ≤ M < N ≤ 10) 둘째 줄에 떨어지는 사과의 개수 J가 주어진다.
(1 ≤ J ≤ 20) 다음 J개 줄에는 사과가 떨어지는 위치가 순서대로 주어진다.
모든 사과를 담기 위해서 바구니가 이동해야 하는 거리의 최솟값을 출력한다.
어렵지 않은 그리디, 구현 문제였지만 바구니의 시작 끝 위치 변화의 구현이 어려워 풀이에 실패했다. 풀이를 보고 아이디어를 얻은 후 다시 작성해보았다.
import sys
inp = sys.stdin.readline
n, m = map(int, inp().split())
j = int(inp())
lis = []
answer = 0
# 바구니 시작과 끝 위치
start = 1
end = m
# 사과 떨어지는 위치들 리스트에 넣어줌
for i in range(j):
lis.append(int(inp()))
for i in range(len(lis)):
# 사과 떨어지는 위치가 바구니 범위 안에 있으면 움직이지 않음
if lis[i] >= start and lis[i] <= end:
continue
# 사과 떨어지는 위치가 바구니 왼쪽보다 앞일 때
if lis[i] < start:
answer += start-lis[i]
end -= start-lis[i]
start = lis[i]
# 사과 떨어지는 위치가 바구니 오른쪽보다 뒤일 때
else:
answer += lis[i]-end
start += lis[i]-end
end = lis[i]
print(answer)