백준 10813문제. 공 바꾸기

문제

문제: 도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다. 도현이는 앞으로 M번 공을 바꾸려고 한다. 도현이는 공을 바꿀 바구니 2개를 선택하고, 두 바구니에 들어있는 공을 서로 교환한다. 공을 어떻게 바꿀지가 주어졌을 때, M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

입력: 첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다. 둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다. 각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다. (1 ≤ i ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 공을 교환한다.

출력: 1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다.

예제 입력: 5 4 

                  1 2

                  3 4

                  1 4

                  2 2

예제 출력: 3 1 4 2 5

 

풀이

const fs = require('fs');
const input = fs.readFileSync(0, "utf-8").trim().split("\n");

const [N, M] = input[0].split(" ").map(Number);
let baskets = Array.from({ length: N }, (_, i) => i + 1);

for (let t = 1; t <= M; t++) {
  const [i, j] = input[t].split(" ").map(Number); 
  [baskets[i - 1], baskets[j - 1]] = [baskets[j - 1], baskets[i - 1]];
}

console.log(baskets.join(" "));

 

풀이 해석

1. 바구니 초기화 let baskets = Array.from({ length: N }, (_, i) => i + 1);

  - 길이가 N인 배열을 만들고, [1, 2, 3, ..., N]으로 초기화 한다.

2. M번 반복하면서 공을 교환 for반복문

  - ES6 구조 분해 할당([a, b] = [b, a])을 사용하여 간단하게 공을 교환한다.

3. 최종 결과 출력

  - join(' ')을 사용하여 배열을 공백으로 출력한다.

 

더 알아보기

1. Array.from({ length: N }, (_, i) => i + 1);의 동작 원리

1-1. Array.from()의 매개변수

1-1-1. 첫 번째 매개변수(value)여기서는 (_)

           - 배열을 생성할 때 기본 값을 의미하지만, {length:N}처럼 객체를 넘기면 이 값은 undefined가 된다.

           - 사용할 필요가 없기 때문에 _로 처리한다.

1-1-2. 두 번째 매개변수(index)여기서는(i)

          - 현재 배열 요소의 인덱스 번호를 나타낸다.

          - 이를 활용해 i+1을 반환하면 [1, 2, 3, ..., N]형태의 배열을 만들 수 있다.

 

2. 사용하지 않는 매개변수라면 ( _ )로 쓰는 것이 관례적이다.