[BOJ 1158 / Python] 요세푸스 문제

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

1158번: 요세푸스 문제

 

 

회전하는 큐 문제라고 생각하면 매우 쉬워진다! 그냥 rotate를 사용하면 끝!

 

다만 파이썬에서 사용하는 출력이 매우 불편했다..... 무조건 줄바꿈이 되는 print는 불편해...

그래서 다른 사람들의 프린트하는 부분의 코드만 봐봤다.

 

from collections import deque

N, K = map(int, input().split())
dq = deque([i+1 for i in range(N)])
arr = []
while(len(dq)>0):
    dq.rotate(-(K-1))
    arr.append(dq.popleft())


print('<', end='')
for i in arr[:-1]:
    print(i, end=', ') 
print(str(arr[-1]) + '>')

print("<", ", ".join(map(str, arr)), ">", sep="")

# 출력하는 부분만