[알고리즘]section 6-6_공주 구하기

Yul·2022년 5월 14일

Algorithm

목록 보기

6/9

문제

공주 구하기

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

이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.
예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자. 처음에는 3번 왕자가 3을 외쳐 제외된다. 이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7번 왕자에게 공주를 구하러갑니다.
N과 K가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.

입력예제
8 3

출력예제
7

문제풀이

1. queue는 Array으로 생성. 배열 초기화를 위해 셋팅해준다. 유사배열 객체
- length: n 1부터 8까지 들어갈 공간을 셋팅. 하지만 이렇게만 해주면 길이만 n인 객체가 생기는 것임.(값은 undefined)
- 화살표 함수를 사용하여 v, i를 인수로 전달. 배열의 value값과 index. 인덱스를 고려하여 i+1로 채우기
2. queue가 하나 있을 때 만큼 돌면 되므로 while문으로 queue의 길이만큼 순회한다. (0이 되면 멈춤)
3. for문으로 queue 배열의 k번째만큼 순회하고 첫번째에 위치한 요소를 queue.push(queue.shift())로 배열의 맨 끝으로 넣음.
4. k번째의 마지막에 있는 요소는 shift()로 제거
5. while문 마지막에 queue배열의 요소가 하나 남았을 때 answer에 남은 요소를 넣어준다.

** for문으로 queue 배열의 초기화를 시켜주는 것과 Array.from()을 이용해 초기화를 시켜주는 것에 대해 질문하신 분이 있었는데 둘 다 시간복잡도의 차이는 없다고 한다.

코드_내가 푼 코드

function solution(n, k) {
  let answer;
  let a = [];

  for(let i = 1; i < n + 1; i++) {
    a.push(i);
  }
  for(let x of a){
    a.splice(k-1, 1);
    answer = x;
  }
  return answer;
}

let a = 6;
let b = 2;
console.log(solution(a, b));

코드_선생님의 코드

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;
}
console.log(solution(8, 3));