백준 2828: 사과 담기 게임 - javascript(그리디)

문제 출처 

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

2828번: 사과 담기 게임

문제

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

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

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

입력

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

출력

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

정답 풀이 힌트

n = 5, m = 2 , j = 1,5,3 일 때

정답 풀이 코드 

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);

function solution(N,M, input){
    let left = 1;
    let right = M; 
    let answer = 0; 
    for(let x of input){
        if(x > right){
            answer += (x - right); 
            right = x;
            left = x - (M-1); 
        }else if(x < left){
            answer += (left - x)
            left = x; 
            right = x + (M-1);
        }
    }
    return answer; 
}
console.log(solution(N,M,input));

의사 코드 추가 

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);

function solution(N,M, input){
    let left = 1; //시작은 맨 왼쪽이다. 
    let right = M; //바구니 크기가 오른쪽 값이 된다. 
    let answer = 0; //바구니 이동 거리 최솟값
    
    //현재 가리키는 위치가 바구니의 왼쪽과 오른쪽 범위에서 벗어날 때만 이동을 한다.  
    for(let x of input){
       //오른쪽보다 클 때 
        if(x > right){ 
            answer += (x - right); 
            right = x;
            left = x - (M-1); //오른쪽의 위치에서 (바구니 크기-1)를 뺀 값이 왼쪽의 위치가 된다.  
        }else if(x < left){ //왼쪽보다 작을 때 
            answer += (left - x)
            left = x; 
            right = x + (M-1);//왼쪽의 위치에서 (바구니 크기-1)를 뺀 값이 오른쪽의 위치가 된다.
        }
    }
    return answer; 
}
console.log(solution(N,M,input));