[자료구조] 공주구하기

 

 

 

설명

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다.
정보 왕국에는 왕자가 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를 이용한다. Queue는 FIFO(First In, First Out) 이다.

 

Queue 사용법은 다음과 같다.

• 큐 선언 : Queue<Integer> q = new LinkedList<>();
• 데이터 넣기 : offer
• 데이터 꺼내기 : poll
• 제일 앞에 있는 값 확인 : peek
• 큐의 크기 : size
• 해당 요소가 포함되어 있는가 : contains

 

큐가 비어있지 않을 동안 반복한다. (while (!queue.isEmpty()))

한 번의 루틴은 다음과 같다.

• 0부터 k-1번째 왕자를 poll 하고 offer 한다. (맨 앞에서부터 꺼내서 뒤에 넣는다는 것)
• k번째는 그냥 poll한다. (k를 외친 사람은 제외되는 것이다.)
• 큐의 사이즈가 1이라면 남아있는 그 값을 큐에서 꺼내어 answer로 한다.

 

 

 

 

코드

package inflearn._5_6;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {

    public static int solution(int n, int k) {

        int answer = 0;

        Queue<Integer> q = new LinkedList<>(); // 큐 선언

        // 초기 데이터 세팅
        for(int i = 1; i <= n; i++) {
            q.offer(i);
        }

        // 핵심 로직
        while(!q.isEmpty()) { // 큐가 비면 멈춤

            for(int i = 1; i < k; i++) {
                q.offer(q.poll()); // 앞에꺼 꺼내서, 맨 뒤에 넣음
            }

            q.poll(); // K를 외친 사람이면 큐에서 빼기

            if (q.size() == 1) answer = q.poll(); // 큐 사이즈가 1일 때 남은 그 값이 정답
        }

        return answer;
    }

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());

        bw.write(sb.append(solution(n, k)).toString());
        bw.flush();

        bw.close();
        br.close();

    }
}