1158번 - 요세푸스 문제

https://www.acmicpc.net/problem/1158

1158번: 요세푸스 문제

처음에는 재귀로 풀었는데 런타임 에러가 떴다.

그래서 재귀를 안 쓰고 푸는 방법을 생각하다가 바로 Q에서 popleft하는 방법이 떠올랐다.

코드도 훨씬 간단해졌다.

# 요세푸스 문제
import sys
from collections import deque

input = sys.stdin.readline
N, K = map(int, input().split())
Q = deque([i for i in range(1, N+1)])
answer = []

while Q:
    for i in range(1, K):
        Q.append(Q.popleft())
    answer.append(Q.popleft())

print('<', end='')
for a in answer[:-1]:
    print(a, ', ', end='', sep='')
print(answer[-1], '>', sep='')