[JAVA]6. 공주 구하기 (Queue)

설명

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

정보 왕국에는 왕자가 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	예시 출력1
8 3	7

 

설명

처음에 문제를 읽고 나서 무슨 소리인지 이해가 잘 안 돼서 여러 번 읽고 생각을 오래 했다.

이 문제는 369 게임처럼 해당하는 K번째 당첨되는 사람을 제거하는 프로세스다.

1번부터 시작해서  K번째 index 숫자를 제거하고, 제거된 숫자 다음 사람부터 K번째에 해당하는 index를 제거하여 마지막에 남은 숫자를 출력하는 문제다.

 

N = 8, K = 3 경우

1. queue를 1~N으로 초기화
2. queue에 아무것도 없을 때까지 반복
    k-1만큼 반복문을 돌면서 k번째 오는 왕자를 제거
        -> 1 2 3 4 5 6 7 8 (K번째 3)
        -> 4 5 6 7 8 1 2 (K번째 6)
        -> 7 8 1 2 4 5 (K번째 1)
        -> 2 4 5 7 8 (K번째 5)
        -> 7 8 2 4 (K번째 2)
        -> 4 7 8 (K번째 8)
        -> 4 7 (queue는 FIFO 구조로 맨 앞의 7보다 먼저 입력된 4 제거)
3. queue size가 1인 경우 answer = queue.poll();

4. queue.isEmpty() 조건 성립되므로 종료된다.

 

public class Main{
    public static int solution(int N, int K){
       int answer = 0;
       Queue<Integer> queue = new LinkedList<>();
       for(int i=1; i<=N; i++) queue.add(i);

       while(!queue.isEmpty()){
           for(int i=1; i<K; i++) queue.offer(queue.poll()); //k-1까지 뒤로 이동, k번째는 poll
           queue.poll();
           if(queue.size() == 1) answer = queue.poll(); //queue size가 1개일 경우 return
       }
       return answer;
    }

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int N = scan.nextInt();
        int K = scan.nextInt();
        System.out.println(solution(N, K));
    }
}