BOJ 1158 - 요세푸스 문제

문제

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

1158번: 요세푸스 문제

 

 

 

풀이

큐를 이용해 문제를 해결할 수 있다.

큐에서 k - 1 번 데이터를 front에서 pop하고 rear에 넣어준다. 이것이 원을 이루면서 앉아있는 것과 같아진다.

그 다음 k 번째에 데이터를 pop한다.

이를 q가 빌 때까지 반복하고 pop한 숫자들을 차례대로 저장해 출력한다.

 

 

 

코드

python

from collections import deque
 
def solution(n, k):
    answer = []
    q = deque(range(1, n + 1))
    while q:
        for _ in range(k - 1):
            q.append(q.popleft())
        answer.append(q.popleft())
    return answer            
 
n, k = map(int, input().split())
answer = solution(n, k)
 
print(str(answer).replace('[','<').replace(']','>'))