문제풀이/백준(Boj) 문제풀이

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

얄루몬 2022. 9. 6. 23:03

from collections import deque

n, k = map(int,input().split())

q = deque(range(1, n+1))

res = []

while q:
    for i in range(k-1):
        q.append(q.popleft())
    res.append(q.popleft())

print("<",end="")
for i in range(n):
    if i == n-1:
        print(f"{res[i]}>", end="")
    else:
        print(f"{res[i]}, ",end="")
print(f"<{', '.join(map(str,res))}>")
  • join을 사용한 출력문의 간소화!