728x90
반응형

순환큐 문제이다. 11866번 요세푸스 문제 0과 같은 문제이다.

11866번은 그냥 리스트로 큐를 구현해서 풀어도 맞았었는데 이 문제는 시간초과가 나와서 

속도가 매우 빠른 collections.deque를 사용했다. 

from collections import deque

N, K = map(int, input().split())
queue = deque([i for i in range(1, N+1)])
res = []
for _ in range(N):
    for i in range(K-1):
        queue.append(queue.popleft())
    res.append(queue.popleft())
print('<', end='')
print(*res, sep=', ', end='')
print('>')
728x90
반응형

+ Recent posts