CODEKATA #35

공주 구하기

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

나의 풀이

function solution(n, k) {
  let answer;
  let circle = [];
  for (let i = 1; i < n + 1; i++) {
    circle.push(i);
  }
  let idx = 1;
  while (circle.length != 1) {
    if (idx % k != 0) circle.push(circle[0]);
    circle.shift();
    idx = idx + 1;
  }
  answer = circle[0];
  return answer;
}

정답 풀이

function solution(n, k){
  let answer;
  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();
  }  
  return answer;
}

Javascript에서의 Queue

Array를 활용하여 Javascript에서 자료구조 Queue를 사용할 수 있다.
Array.prototype.push(i)를 통해 enqueue(i)를 ,
Array.prototype.shift()를 통해 dequeue()를 구현할 수 있다.

마지막 왕자(예시에서의 8번 왕자)가 숫자를 외친 후, 첫번째 왕자가 다음 숫자를 다시 외치기 때문에 원형 큐를 사용하여 특정 조건(한명의 왕자만 남는다!)이 만족될 때까지 반복하도록 했다.

원형 큐를 구현한 방식은 각각의 왕자가 제외되지 않는다면, 큐에 다시 추가되도록 하여 8 이후 1이 들어가도록 했다.
예를 들어, 첫번째 Queue에서 저 시점의 4번 왕자는 제외되는 조건이 아니기 때문에 다시 Queue에 추가된다.