문제 설명

문제 링크는 여기.

1158번: 요세푸스 문제

원이라고 나와있지만, 큐에 담긴 N명의 사람들을 K번 점프할 때마다 뽑아 출력하면 되는 문제이다.

접근

단순하게 K번 점프할 때마다 큐의 원소를 뽑아 출력하는 형태로 구현했다.

sb.append("<");

// 일단 순서대로 넣고
for(int n=1; n<=N; n++) {
	queue.add(n);
}

// 큐가 빌때까지 반복
while(!queue.isEmpty()) {
	// k번 점프
	for(int k=0; k<K-1; k++) {
		queue.add(queue.poll());
	}
	// 하나 출력
	sb.append(queue.poll() + ", ");
}
sb.setLength(sb.length()-2);
sb.append(">");
System.out.println(sb);

소스코드

최종 제출한 코드는 다음과 같다.

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

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		Queue<Integer> queue = new LinkedList<>();
		String[] input = br.readLine().split(" ");
		int N = Integer.parseInt(input[0]);
		int K = Integer.parseInt(input[1]);
		
		sb.append("<");
		
		for(int n=1; n<=N; n++) {
			queue.add(n);
		}
		while(!queue.isEmpty()) {
			for(int k=0; k<K-1; k++) {
				queue.add(queue.poll());
			}
			sb.append(queue.poll() + ", ");
		}
		sb.setLength(sb.length()-2);
		sb.append(">");
		System.out.println(sb);
	}
}

결과

제출 결과는 다음과 같았다.

접근

같은 코드로 요세푸스 문제 0도 해결된다.

추가적으로 요세푸스 문제 3도 풀어보고자 했으나 실패하였다… 다음에 다시 시도해보고자 한다.