[Python] 요세푸스 문제 - 백준

해당 글은 백준 1158번 문제 '요세푸스 문제'를 다룬다.

문제

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

1158번: 요세푸스 문제

 

설명

deque의 rotate 함수를 사용해서 -(k-1)만큼 큐를 돌리고, 가장 첫번째 값만 계속 뽑으면 된다.

그리고 출력을 할 땐, replace 함수를 이용해서 리스트의 [ ] 괄호를 < > 괄호로 변경할 수 있다.

 

코드

from collections import deque

n, k = map(int, input().split())
queue = deque([i for i in range(1, n+1)])

result = []

for i in range(n):
    queue.rotate(-(k-1))
    first_value = queue.popleft()
    result.append(first_value)

print(str(result).replace('[','<').replace(']','>'))