[Python] 백준 #1158. 요세푸스 문제

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

1158번: 요세푸스 문제

 

# 1158 - Python 3

import sys

input = sys.stdin.readline

n, k = map(int, input().split())

arr = [i for i in range(1, n + 1)]
answer = []

idx = 0

for _ in range(n):
    idx += k - 1                                # pop될 arr 값의 idx
    
    if idx >= len(arr):                         # idx가 남은 수의 개수보다 큰 경우
        idx %= len(arr)
    
    answer.append(str(arr.pop(idx)))            # join 함수는 str형태에 적용됨

print("<", ", ".join(answer), ">", sep = '')