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

문제

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

1158번: 요세푸스 문제

설계

파이썬

N, K = map(int, input().split())
arr = [i + 1 for i in range(N)]

answer = []
now = K - 1
while len(arr):
    answer.append(arr[now])
    arr.pop(now)
    now = now + K - 1       # pop을 먼저 했으므로 숫자 1개가 줄어든다.
    if len(arr) > 0:        # zero division 방지
        now %= len(arr)

# 결과 출력
print('<', end='')
for i in range(len(answer)):
    if i == len(answer) - 1:
        print(answer[i], end='')
    else:
        print(answer[i], end=', ')
print('>', sep='')