코테 대비 python/백준

1158 요세푸스

ylab 2023. 2. 6. 19:36
 

1158번: 요세푸스 문제

첫째 줄에 N과 K가 빈 칸을 사이에 두고 순서대로 주어진다. (1 ≤ K ≤ N ≤ 5,000)

www.acmicpc.net

####시간 84ms
####메모리 34244KB
#rotate??
#import sys
#input = sys.stdin.readline
from collections import deque
n, k = map(int,input().split())
a= [_+1 for _ in range(n)]
queue = deque(a)
result = []
while queue:
    #왼쪽으로 돌려야해 
    queue.rotate(-k+1)
    b= queue.popleft()
    result.append(b)

print("<",end="")
for _ in result:
    if _ == result[len(result)-1]:
        print(f"{_}>")
    else:
        print(f"{_},",end=" ")
 
rotate 함수 활용