[ 2023-03-03 🐖 TIL ]

백준 1158번

1 ~ N 까지의 수를 원형을 만들어서 k번째 수를 추출하는 문제입니다.
(k번째의 수가 추출이 되면 다음 인덱스인 k+1부터 다시 k번째의 수를 추출합니다.)

import sys

input = sys.stdin.readline

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

li = [i for i in range(1, n+1)]
result = []
while len(li) >= k:
    result.append(li[k-1])

    li = li[k:] + li[0: k-1]

#나머지 k보다 작은 길이의 배열이 남았을 때
for i in range(len(li)):
    index = 0
    for _ in range(k-1):
        if index == len(li)-1:
            index = 0
        else:
            index += 1

    result.append(li[index])
    li = li[index+1:] + li[0: index]


result = result + li


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

문제를 잘못이해해서 푸는 방식을 알아보고 내 나름대로 풀어서 통과가 되었습니다. :)