백준 1158 요세푸스 문제(JAVA)

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

1158번: 요세푸스 문제

 

 

문제)

 

 

 

풀이)

 

Queue를 생성하여 모든 원소를 넣어두고,

pop한 후 순서에 맞춰 버릴지 queue에 다시 넣을지 결정한다!

이때, 순서에 맞춰 버리는 애는 순서대로 정답 배열에 저장해두고 후에 출력하면 끝!

 

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Main_BJ_1158_요세푸스문제 {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int k = sc.nextInt();
		Queue<Integer> p = new LinkedList<>();
		int[] answer = new int[N];
		for(int i=1; i<=N; i++)
			p.offer(i);
		
		int count=1;
		int index=0;
		while(!p.isEmpty()) {
			if(count % k == 0) {
				answer[index] = p.poll();
				index++;
				count++;
				continue;
			}
			p.offer(p.poll());
			count++;
		}
		System.out.print("<");
		for(int i=0; i<N; i++) {
			if(i!=N-1)
				System.out.print(answer[i] + ", ");
			else
				System.out.print(answer[i]);
		}
		
		System.out.println(">");
		sc.close();
	}

}