11866번 : 요세푸스 문제0

문제보기

파이썬

from collections import deque


n, k = map(int, input().split())  # n: 인원수, k : 몇번째 사람 제거
arr = []
answer = []

for i in range(1, n+1):
    arr.append(i)

queue = deque(arr)

while queue:
    queue.rotate((-1*k)+1)
    answer.append(queue.popleft())

print('<'+str(answer)[1:-1] + '>')

코드 설명

• 배열의 0번 인덱스만 빼주기 위해 queue로 만들어주기
• rotate() 메서드를 이용해 큐안의 값 움직이기

포인트

코드를 구현하는거보다 마지막에 print하는 부분에서 어떻게 해야할지 몰라 진땀 뺏다..