Algorithm/BaekJoon
[BOJ] 1158. 요세푸스 문제
히욤이
2020. 2. 8. 03:12
BaekJoon 1158. 요세푸스 문제
* 문제의 저작권은 BOJ 및 문제를 만든 사람에게 있습니다.
[문제 접근]
Queue를 이용해서 K번만큼 for문을 돌려서 큐에 빼고 넣고 반복하고 K번째에는 빼서 출력
처음에는 break문을 while문 맨 아래에 적었더니 출력초과라고 나왔다.
백준 풀면서 출력초과는 처음 봐서 찾아봤더니 출력 값이 많아서 그렇다고 한다.
예외를 생각해봐도 생각이 안나서 그냥 if문을 while문 맨 위로 수정했더니 통과가 되었는데 왜 그런지 모르겠다.
근데 방금 이 글을 쓰면서 생각해봤는데 탈출문이 아래에 있어서 1,1을 넣었을 때는 <1>이 출력되는게 아니라
무한 NULL이 출력되서 그렇더라
[코드]
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
public class Main_1158 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int N = sc.nextInt();
int K = sc.nextInt();
Queue<Integer> queue = new LinkedList<>();
for (int i = 1; i <= N; i++) {
queue.add(i);
}
System.out.print("<");
while (true) {
if (queue.size() == 1) {
System.out.print(queue.poll());
break;
}
for (int i = 1; i <= K; i++) {
if (i == K) {
System.out.print(queue.poll()+", ");
} else {
queue.add(queue.poll());
}
}
}
System.out.print(">");
}
}