[백준/Node.js] 1157번. 바구니 순서 바꾸기 || 풀이 JavaScript

const input = require('fs').readFileSync('/dev/stdin').toString().split("\n").map(v => (v.replace("\r", "")));
const arr = input.map(v => v.split(" "));
const N = arr[0][0];
const M = arr[0][1];

let baskets = [];
let baskets2 = [];
for (let i = 1; i <= N; i++) {
    baskets.push(i);
    baskets2.push(i);
}
const rotateArr = (start, mid, end) => {
    let rotateNum = [];
    rotateNum = baskets.splice(start - 1, end - start + 1);
    const originMidIndex = baskets2[mid - 1];
    const newMidIndex = rotateNum.findIndex(v => v === originMidIndex);
    let midNums = rotateNum.splice(newMidIndex, rotateNum.length - newMidIndex);
    let finishedRotateNums = [...midNums, ...rotateNum];

    baskets2.splice(start - 1, end - start + 1, ...finishedRotateNums);
    return baskets2;
}

for (let i = 1; i <= M; i++) {
    let start = +arr[i][0];
    let end = +arr[i][1];
    let mid = +arr[i][2];

    baskets = rotateArr(start, mid, end).map(v => v);
    baskets2 = baskets.map(v => v);
}
console.log(...baskets);

 

어떻게 해야할지 생각이 안나고, 잘 안돼서 풀다 한번 포기했었다. 🤨😫🤯

 

 

그러다 오늘은 splice 함수가 생각났다. 

https://developer.mozilla.org/ko/docs/Web/JavaScript/Reference/Global_Objects/Array/splice

Array.prototype.splice() - JavaScript | MDN

 

 

<splice 함수 사용하는 방법>

배열.splice( a , b, c );

a = 시작index

b = 잘라낼 개수 (0을 쓰면 어떤 요소도 제거되지 않음)

c = 잘라낸 부분에 넣을 요소 (안쓰면 제거하기만 함)

return 값: 잘라낸 배열.

splice 한 배열은 변경된 상태.

 

예시)

const food = ['🍕', '🍟', '🥗', '🍢'];
food.splice(1, 0, '🏀');
// index 1 부터 0개 삭제, 그 자리(index 1 자리)에 🏀 을 넣음
console.log(food);
// ["🍕", "🏀", "🍟", "🥗", "🍢"] 배열 변화됨. 
food.splice(1, 3);
// [ '🏀', '🍟', '🥗'] //return 값임. 바로 위의 배열에서 index 1 부터 3개 도려냄
console.log(food);
// [ '🍕', '🍢' ] //남은 배열 food 의 상태

 

 

 

<풀이 과정>

전체 바구니 함수에서 splice 함수를 이용해 회전 시킬 부분을 먼저 가져온 후,

가져온 배열을 mid 를 중심으로 분리해준 다음에 합쳐줬다.

 

이 때, 주어진 mid 의 수는 원래 함수에서의 N번 째 바구니이기 때문에, 

원래 baskets 배열에서, index mid -1 이 어떤 숫자인지 찾은 후, 

회전 시키려고 떼어준 배열에서 그 수의 index를 찾아 중심으로 삼았다. 

 

baskets과 똑같은 배열 baskets2 를 만들어줬다.

그냥 할당시 얕은 복사가 일어나기 때문에 map 함수로 새로운 함수를 할당해줬다. (같은 메모리 주소를 가리키고 있음)

 

for 문으로 이 모든 과정을 거친 후, 

가장 마지막에 나오는 baskets 배열을 펼쳐서 출력했다.

 

풀어내서 매우 속시원하다!