공주 구하기

설명

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다.

정보 왕국에는 왕자가 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)가 주어진다.

 

출력

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

 

예시 입력 1 

8 3

 

예시 출력 1

7

 

 

문제 해결

 

Queue는 FIFO 구조로 먼저 들어간 것이 먼저 나오는 구조이다. 여기에서는 특정 숫자에 해당하면 빠지고, 그 이후부터는 다시 1부터 세는 방식으로, 특정 숫자의 전까지는 큐의 뒤로 다시 넣는다고 생각하면 된다.

 

우선 총 인원 N을 입력받고, 특정 숫자 K를 입력 받은 후 Integer 형의 큐를 만들어준다.

Scanner in = new Scanner(System.in);
int N = in.nextInt();
int K = in.nextInt();
Queue<Integer> queue = new LinkedList<>();

 

그리고 num 변수를 1로 초기화해 만들어주었고, N까지 돌며 큐에 num을 하나씩 증가시키며 넣어주었다. 즉 큐에는 1부터 N까지의 숫자가 차례로 들어가게 되는 것이다.

int num = 1;
for(int i = 0; i < N; i++){
    queue.offer(num++); // 큐에 1번부터 N번까지 담음
}

 

 

특정 숫자 K보다 작으면 큐에서 빼고 다시 큐의 뒤로 넣어준다. 그리고 K번째는 제거해준 후 다시 for문을 수행해준다. 마지막에 남은 왕자를 출력하라고 했으므로 큐의 사이즈가 1보다 큰 동안 이 과정을 반복해주면 된다.

while (queue.size() > 1) { // 큐에 1명이 남을 때까지
    for (int i = 1; i < K; i++) {
        queue.offer(queue.poll());  // K번째 전까지는 다시 큐의 뒤로 보냄
    }
    queue.poll();  // K번째는 제거
}

 

그리고 큐의 사이즈가 1이 되어 while문을 빠져나오면 그 때 queue.poll() 을 출력해주면 된다.

System.out.println(queue.poll());

 

 

최종 코드

 

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class P06_공주구하기 {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        int K = in.nextInt();
        Queue<Integer> queue = new LinkedList<>();

        int num = 1;
        for(int i = 0; i < N; i++){
            queue.offer(num++); // 큐에 1번부터 N번까지 담음
        }

        while (queue.size() > 1) { // 큐에 1명이 남을 때까지
            for (int i = 1; i < K; i++) {
                queue.offer(queue.poll());  // K번째 전까지는 다시 큐의 뒤로 보냄
            }
            queue.poll();  // K번째는 제거
        }

        System.out.println(queue.poll());
    }
}