[백준]11866번 요세푸스 문제 0 - 큐

 

from collections import deque
queue = deque([])
n, k = map(int, input().split())
result = []
for i in range(n):
    queue.append(i+1)

while True:
    if len(queue) == 0 :
        break
    for i in range(k-1):
        queue.append(queue.popleft())
    result.append(queue.popleft())

print('<'+', '.join(map(str, result))+'>')

 

코드 설명

반복문을 돌면서 큐의 길이가 0 인 경우 종료 한다. 반복문 안에서 K의 값 만큼 반복문을 수행하는데, 첫 번째 원소의 자리를 바꾸기 위해 popleft() 한 값을 append() 한다. K 번의 반복이 끝난 후 결과 리스트에 queue.popleft() 한 값을 넣어준다.

결과 형식에 맞게 join 함수를 통해 출력하여 준다.

 

Point : 큐에서 popleft() 한 값을 다시 뒤로 넣는 과정을 K-1번 반복하는 것이 중요하다. (원의 형태를 이루고 있으므로 제거하지 않은 사람은, 큐에서는 뒤로 가는 것과 동일) 그리고 결과값은 따로 저장한다.

형식에 맞게 출력하기 위해 join() 함수를 이용한다.