알고리즘 : 공주구하기 (인프런, 큐를 이용한 문제)

 

인프런 JS 알고리즘 강좌 문제를 한번 풀이해보겠습니다.

 

강좌는 아래 인프런 강좌를 참고해주시면 됩니다.

 

강좌 바로가기

 

 

문제 설명

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

 

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

 

입력 설명

첫 줄에 자연수 N(5<=N<=1,000)과 K(2<=K<=9)가 주어진다.

 

출력 설명

첫 줄에 마지막 남은 왕자의 번호를 출력합니다.

입력 예제

8 3

출력 예제

7

 

문제 해석

 

먼저 문제를 이해하기 위해서는 3, 6, 9 게임이라고 생각하면 이해가 쉽습니다. 

 

왕자들이 앉은 순서대로 번호를 외치다가 3을 외친 왕자는 제외되는 것입니다. 

 

여기서의 3은 입력받은 K를 말합니다. 

 

만약 3을 외친 왕자가 빠지면, 4번 왕자부터 다음 게임을 진행합니다.

 

문제는 이해를 했지만 어떻게 풀어야 하는지 몰라서 고민을 많이 했습니다.

 

 

노트 메모

 

 

내 해답

function mySolution(n, k){
  let answer;
  const queue = Array.from(new Array(n), (x, i) => i + 1)

  for (let i =1; i<= n.length; i++) {
    queue.push(n[i])
  }

  let count = 0;
  while(queue.length !== 1) {
    count++
    if (count === k) {
      count = 0
      queue.shift()
    } else {
      const item = queue[0]
      queue.shift()
      queue.push(item)
    }
  }
  answer = queue[0]

  return answer;
}

1. 입력한 문자열에 대한 배열을 for 문을 돌면서 숫자인지 아닌지 판단합니다. 
2. 숫자면 Stack 에 push 합니다.
3. 숫자가 아닌 문자(연산자)가 들어오면 Stack 에서 pop 을 한 숫자를 연산자를 통해 계산합니다. 
   이때 맨 처음 계산일 경우 pop 을 2번하여 숫자 2개를 꺼내어 계산하고, 이후에는 result 와 pop 한 숫자 1개를 꺼내서 계산합니다.
4. 해당 결과를 Result 에 저장합니다.
5. Result 를 반환합니다.

 

 

인프런 해답

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;
}

1. 연산자가 아닌 숫자를 Stack 에 push 합니다.
2. 숫자가 아니라면 Stack 에서 Pop 을 2번 수행합니다.
3. 이때 먼저 pop 한 숫자를 rt 나중에 pop 한 숫자를 lt 로 변수를 할당합니다.
4. 연산자에 맞게 계산하고 결과를 다시 Stack 에 push 합니다.
5. 마지막에 stack[0] 만 남게 되는데 해당 값을 반환합니다.

 

 

마무리

 

해당 문제를 풀면서 eval 이라는 Javascript 함수를 알수 있었습니다.
(문자열에 연산자가 있으면 계산후에 반환해주는 함수)

 

또 Javascript에서 Stack 사용에 대해 이해할 수 있었습니다.

 

제가 알고리즘쪽 공부를 많이 해보지 않아서 조금 서툴지만 프론트엔드 개발자이긴 해도 알고리즘 공부를 해두면 좋을것 같아서 다시 공부를 시작하고 있습니다.

 

앞으로도 블로그에 꾸준히 알고리즘 관련 포스팅을 써볼 예정입니다.