[백준] 1158번 요세푸스 문제 - 자바 Queue 사용

문제

요세푸스 문제는 다음과 같다.

1번부터 N번까지 N명의 사람이 원을 이루면서 앉아있고, 양의 정수 K(≤ N)가 주어진다. 이제 순서대로 K번째 사람을 제거한다. 한 사람이 제거되면 남은 사람들로 이루어진 원을 따라 이 과정을 계속해 나간다. 이 과정은 N명의 사람이 모두 제거될 때까지 계속된다. 원에서 사람들이 제거되는 순서를 (N, K)-요세푸스 순열이라고 한다. 예를 들어 (7, 3)-요세푸스 순열은 <3, 6, 2, 7, 5, 1, 4>이다.

N과 K가 주어지면 (N, K)-요세푸스 순열을 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 K가 빈 칸을 사이에 두고 순서대로 주어진다. (1 ≤ K ≤ N ≤ 5,000)

 

출력

예제와 같이 요세푸스 순열을 출력한다.

 

---

 

Queue클래스를 사용해 문제를 풀어보았다.

풀이

1. 입력받은 N을 이용해 1부터 N까지의 수를 Queue에 add하였다.
2. K번 반복하며 앞에서부터 삭제하고 다시 뒤에 추가해준다.
3. 그리고 제일 앞에 남은 데이터를 출력한다.

앞에서 삭제하고, 뒤에 추가, 앞의 데이터 출력 → 선입선출의 큐와 같은 모습으로 진행 된다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class BOJ_1158 {
    static Integer N;
    static Integer K;
    static Queue man;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        N = Integer.parseInt(st.nextToken());
        K = Integer.parseInt(st.nextToken());

        man = new LinkedList();

        for (int i = 1; i < N+1; i++) {
            man.add(i);
        }


        System.out.print("<");
        while(!man.isEmpty()){
            for (int j = 0; j < K-1; j++) {
                man.add(man.remove());
            }

            System.out.print(man.peek());
            man.remove();

            if(man.isEmpty())
                break;
            else System.out.print(", ");
        }
        System.out.print(">");
    }

}

 

#1.
Queue는 LinkedList로 구현하여 man이라는 이름의 큐를 생성하였다.
1부터 시작하여 N+1로 반복하고, 반복변수를 이용해 큐에 값을 채워넣었다.

 

#2.
while을 사용해 큐가 비어있지 않을 때까지 앞에서부터 K-1씩 반복한다.
j를 1부터 시작해 K까지 반복해도 된다. 인덱스가 0부터 시작하여 K-1까지 반복하였다.

Queue.remove()를 통해 앞에 들어간 데이터를 삭제하면서 동시에 Queue.add()로 뒤에 삽입하였다.

 

#3.
K번 반복이 끝나면 큐의 제일 앞에 있는 데이터를 Queue.peek()을 통해 출력한다.
출력한 데이터는 remove로 삭제한다.

 

#4.
출력 형식을 동일하게 하기 위해 데이터 출력 후 ", "를 출력하였다.
제일 마지막 데이터를 출력한 뒤에 쉼표를 추가해줄 필요가 없어 데이터 출력 후 다시 한 번 Queue가 비어있는지 확인하였다.
Queue가 비어있으면 바로 break를 하고, 비어있지 않으면 다음 데이터 출력을 위해 쉼표를 출력해주었다.

while문을 감싸고있는 괄호 출력도 모두 출력 형식을 위해 추가하였다.