문제
상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 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문제풀이
💡 문제풀이 과정
- 가장 처음에 바구니는 왼쪽 M칸을 차지하고 있다고 한다. 따라서 바구니의 왼쪽 좌표와 오른쪽 좌표를 정해주는데, 왼쪽 좌표는 1, 오른쪽 좌표는 M이다.
- 사과가 떨어지는 위치값을 반복문으로 돌면서, 사과가 떨어지는 위치가 바구니의 바깥쪽 왼편에 떨어질 때와, 오른편에 떨어질 때를 나누어서 2가지 조건문을 세운다.
- 조건문 안에서 움직임 수와 바구니 내부 왼쪽, 오른쪽의 좌표를 갱신한다.
- 예제 2번 그림 설명 ▼
✅ 답안
const filePath = process.platform == 'linux' ? '/dev/stdin' : './input.txt';
let [n, ...input] = require('fs').readFileSync(filePath).toString().trim().split('\n');
const [N, M] = n.split(' ').map(Number);
const J = Number(input.shift());
input = input.map(Number);
let left = 1; // 바구니 내부의 왼쪽 좌표 초기값
let right = M; // 바구니 내부의 오른쪽 좌표 초기값
let answer = 0; // 움직임 수 저장할 변수
for (const pos of input) {
// pos: 사과의 위치
if (right < pos) { // 사과의 위치가 바구니보다 오른쪽에 떨어짐
answer += pos - right;
right = pos;
left = pos - (M - 1);
} else if (left > pos) { // 사과의 위치가 바구니보다 왼쪽에 떨어짐
answer += left - pos;
left = pos;
right = pos + (M - 1);
}
}
console.log(answer);