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

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

1158번: 요세푸스 문제

 

 

Python

#요세푸스 문제

from collections import deque

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

step = 1
while arr:
    print(arr)
    if step == k:
        ans.append(str(arr.popleft()))
        step = 1
        continue
    step+=1
    arr.append(arr.popleft())
    
print('<'+', '.join(ans)+'>')