[백준] 11866번: 요세푸스 문제 0(Python)

문제 

 

11866번: 요세푸스 문제 0

 

---

 

제출 코드 

from collections import deque

N, K = map(int, input().split())
q = deque()

for i in range(1, N+1):
    q.append(i)

li = []
while q:
    for i in range(K-1):
        q.append(q.popleft())
    li.append(q.popleft())

print("<", end="")
for i in range(len(li)-1):
    print("%d" % li[i], end=", ")
print(li[-1], end="")
print(">")

 

이 문제는 deque를 사용하면 되는 문제이다.

popleft()는 가장 왼쪽의 원소를 제거해주며 제거한 값을 return 해준다.

따라서 위에 q.append(q.popleft())는  '가장 왼쪽 원소를 제거하여 return 한 후, return 된 값을 맨 뒤에 추가한다'는 의미이다.

 

이런 방식으로 K번에 1번 꼴로 원소를 제거하고 출력할 새로운 리스트에 순서대로 추가해주면 된다.