[백준 | Java] 1158 요세푸스 문제

내 풀이

큐를 사용해 1부터 N까지의 수를 add 한 후
poll해서 K의 배수가 아닌 값은 다시 add하고,
K의 배수를 뽑은 값만 sb에 추가한다.

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

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

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

        Queue<Integer> queue = new LinkedList<>();

        for (int i = 0; i < N; i++) {
            queue.add(i+1);
        }
        sb.append('<');

        while (queue.size() > 1) {

            //K의 배수가 아닌 값은 뽑고 다시 넣기
            for (int i = 0; i < K - 1; i++) {
                int val = queue.poll();
                queue.add(val);
            }

            //K의 배수로 뽑은 값만 String에 추가
            sb.append(queue.poll()).append(", ");
        }

        sb.append(queue.poll()).append('>');
        System.out.println(sb);
    }
}