백준 3003문제. 킹, 퀸, 룩, 비숍, 나이트, 폰

문제

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

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

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

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

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

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

예제 입력: 0 1 2 2 2 7 또는 2 1 2 1 2 1

예제 출력: 1 0 0 0 0 1 또는 -1 0 0 1 0 7

 

풀이

const fs = require('fs');
const input = fs.readFileSync(0, 'utf-8').trim().split(' ').map(Number);

const a = [1, 1, 2, 2, 2, 8]; // 올바른 체스 말의 개수
let result = [];

for (let i = 0; i < input.length; i++) {
  if (input[i] !== a[i]) { // 올바른 체스 말의 개수와 같은 개수가 아니라면 빼기연산 후 result에 넣는다.
    result.push(a[i] - input[i]);
  } else { // 올바른 체스 말의 개수와 같은 개수라면 0을 넣는다.
    result.push(0);
  }
}

console.log(result.join(' ')); // 결과 출력

 

풀이 해석

1. a 변수에 올바른 체스 말의 개수를 배열 형태로 할당한다.

2. result 변수에 빈 배열을 할당한다.

3. for문을 돌면서 if문에서 입력받은 input의 배열과 올바른 체스 말의 개수인 a를 인덱스마다 비교한다.

   - 비교하면서 같은 수가 아니라면 빼기연산(-)을 진행하여 값을 result 배열에 추가한다.

   - 같은 수라면 0값을 result 배열에 추가한다.

   - 이때, 배열의 끝에 추가하는 메서드인 push()를 사용한다.

4. 배열형태인 result변수를 String으로 바꾸기위해 join(' ')메서드를 사용하여 백준 문제의 출력에 맞춘다.

5. 참고로 for문은 아래형태처럼 사용하면 코드를 줄일 수도 있다.

for (let i = 0; i < input.length; i++) {
  result.push(correct[i] - input[i]);
}

 

더 알아보기

1. 다른 정답

const fs = require('fs');
const input = fs.readFileSync(0, 'utf-8').trim().split(' ').map(Number);

const correct = [1, 1, 2, 2, 2, 8];
const result = correct.map((val, i) => val - input[i]);

console.log(result.join(' '));

- 좀 더 최적화 되고 간단하게 표현 되었다.