[백준] 3003번: 킹, 퀸, 룩, 비숍, 나이트, 폰 -JS

문제

동혁이는 오래된 창고를 뒤지다가 낡은 체스판과 피스를 발견했다.

체스판의 먼지를 털어내고 걸레로 닦으니 그럭저럭 쓸만한 체스판이 되었다. 하지만, 검정색 피스는 모두 있었으나, 흰색 피스는 개수가 올바르지 않았다.

체스는 총 16개의 피스를 사용하며, 킹 1개, 퀸 1개, 룩 2개, 비숍 2개, 나이트 2개, 폰 8개로 구성되어 있다.

동혁이가 발견한 흰색 피스의 개수가 주어졌을 때, 몇 개를 더하거나 빼야 올바른 세트가 되는지 구하는 프로그램을 작성하시오.

입력 

첫째 줄에 동혁이가 찾은 흰색 킹, 퀸, 룩, 비숍, 나이트, 폰의 개수가 주어진다. 이 값은 0보다 크거나 같고 10보다 작거나 같은 정수이다.

출력

첫째 줄에 입력에서 주어진 순서대로 몇 개의 피스를 더하거나 빼야 되는지를 출력한다. 만약 수가 양수라면 동혁이는 그 개수 만큼 피스를 더해야 하는 것이고, 음수라면 제거해야 하는 것이다.

예제 입력1

0 1 2 2 2 7

예제 출력1

1 0 0 0 0 1

예제 입력2

2 1 2 1 2 1

예졔 출력2

-1 0 0 1 0 7

 

 

 

이 문제는 조금만 생각해 봐도 어떻게 풀어야할지 감이 잡혔다. 하지만 글을 쓰는 이유는 내가 처음에 작성한 코드는 너무 길었다. 

처음 코드

let fs=require('fs');
let input=fs.readFileSync('input.txt').toString().split(' ').map(Number);

let original=[1,1,2,2,2,8];
let result=[];

for(let i=0; i<input.length;i++){
    if(input[i]<original[i]){
        result[i]=original[i]-input[i];
    }else if(input[i]>original[i]){
        result[i]=-(input[i]-original[i]);
    }else{
        result[i]=0;
    }
}
console.log(result.join(' '));

 

for문도 들어가고 조건문도 무수히 많지요.... 위 코드로 정답을 맞추긴 했는데 조금 더 생각해보니 map을 써서 하면 간단해질거라고 생각했다.

 

수정한 코드

let fs=require('fs');
let input=fs.readFileSync('input.txt').toString().split(' ').map(Number);

let original=[1,1,2,2,2,8];
let result=original.map((a,b)=>a-input[b])
console.log(result.join(' '));

 

아직도 map보다 많이쓴 for문이 더 익숙한거 같다. JS로 알고리즘 많이 풀어보면서 map 메소드랑 친하게 지내려고 노력중...💦