[백준 11866] 요세푸스 문제 0 with Python

문제 링크

www.acmicpc.net/problem/11866

11866번: 요세푸스 문제 0

풀이

import sys

n, k = map(int, sys.stdin.readline().split())
lst = [x for x in range(1, n+1)]
permutation = []

idx = 0
while lst:
    idx += k - 1
    if idx >= len(lst):
        idx %= len(lst)
    deleted = lst[idx]
    lst.remove(deleted)
    permutation.append(deleted)

print("<" + ", ".join(map(str, permutation)) + ">")

1부터 n까지 담긴 리스트를 만들어서 k번째 수를 제거하고, 제거된 다음 수부터 시작하여 또 k번째 수를 제거하기를 반복한다. 제거된 수는 빈 리스트인 permutation에 차례대로 저장된다. idx가 지정된 범위를 초과하였을 시에는 남은 수의 개수로 나눈 나머지를 지정한다.