🍀TIL🍀[JavaScript] Coding Test 스터디

자료구조(스택, 큐)

Q. 공주 구하기

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔다.
정보 왕국에는 왕자가 N명이 있는데 서로 공주를 구하러 가겠다고 한다.
정보왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했다.
왕은 왕자들을 나이 순으로 1번부터 N번까지 차례로 번호를 매긴다.
그리고 1번 왕자부터 N번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다.
그리고 1번 왕자부터 시계방향으로 돌아가며 1부터 시작하여 번호를 외치게 한다.
한 왕자가 K(특정숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 된다.
그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다.
이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.

예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자.
처음에는 3번 왕자가 3을 외쳐 제외된다.
이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7번 왕자에게 공주를 구하러 간다.

N과 K가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.

💠solution code

const solution = (n, k) => {
  let answer = 0;
  let queue = Array.from({ length: n }, (v, i) => i + 1);

  while (queue.length) {
    for (let i = 1; i < k; i++) {
      queue.push(queue.shift());
    }
    queue.shift();
    if (queue.length === 1) answer = queue.shift();
    }
  }
};
console.log(solution(8, 3));

• 후입선출이 필요한 문제라서 stack 아닌 queue 사용!

• Array.from({ length: n }, (v, i) => i + 1)

  • from 을 사용해서 1부터 n까지의 배열을 생성한다.
    [1, 2, 3, 4, 5, 6 ... n]

• queue에 왕자들이 남아 있으면 계속해서 while문을 돈다.

  • while문에 들어오면 1부터 k만큼 for문을 돌면서 큐의 가장 앞에 있는 요소를 shift()로 빼내서 push()로 다시 뒤로 넣는다(각자 숫자 외치고 뒤로 빠지는 과정)

  • k - 1 명이 각자의 숫자를 외치고 뒤로 가서 다시 줄 서는 것 끝내면 이제 k를 외치는 왕자가 등장했으니 그 왕자는 shift()해서 빼준다. 공주를 구하러 갈 왕자 리스트에서 제외 되고 원 밖으로 빠진 것.

  • k를 외친 왕자 빼고 난 뒤, 큐의 길이가 1이라면 현재 큐에 마지막 왕자만 남아있다는 것! 따라서 answer의 값을 큐에 마지막으로 남은 왕자의 번호로 설정한다.