백준 1158번 - 요세푸스 (Java 8)

https://www.acmicpc.net/problem/1158

1158번: 요세푸스 문제

큐 활용

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        Queue<Integer> queue = new LinkedList<>();
        StringBuilder sb = new StringBuilder();
        int N = Integer.parseInt(st.nextToken());
        int K = Integer.parseInt(st.nextToken());

        sb.append("<");

        for(int i=0; i<N; i++){
            queue.offer(i+1);
        }
        while(queue.size() != 1){
            for(int i=0; i<K-1; i++){
                queue.offer(queue.poll());
            }
            sb.append(queue.poll()).append(", ");
        }
        sb.append(queue.poll()).append(">");
        System.out.println(sb);
    }
}

반복문을 여러바퀴 돌릴 생각하고 있었는데, 검색해보니 되게 쉽게 푸셔서 따라 풀었다.

 

큐에서 K-1번째 값은 뽑아서 후순위로 보내고, K번째를 출력하는 방식이다. size가 1일때는 굳이 반복문을 K-1번 해줄 필요가 없어서 1까지로 쓰신 것 같다.

 

참고 : 

https://steady-coding.tistory.com/21

[BOJ] 백준 1158번 : 요세푸스 문제 (JAVA)