[백준 BOJ - 1158 / 파이썬] 요세푸스 문제

🔍 문제 분석

원을 이루며 있는 원소들의 k 번째 값을 순서대로 제거하는 문제

 

❗ 해결 아이디어 

k번 째 값을 셀 때 자기 자신도 포함하므로, step = k-1

현재 위치current에서 step을 더한 위치의 값을 반환함

current + step이 리스트의 길이보다 클 경우에는 modular연산을 통해 다시 앞으로 

✔️ 최종 풀이

import sys

n, k = map(int, sys.stdin.readline().split())

josephus = [i for i in range(1, n+1)]
result = []
current = 0
step = k-1

for i in range(n):
    current += step
    if current +1 >= len(josephus):
        current = current % len(josephus)
    result.append(josephus.pop(current))

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