(백준 1158번 문제 풀이) 요세푸스 문제

문제
요세푸스 문제는 다음과 같다.1번부터 N번까지 N명의 사람이 원을 이루면서 앉아있고, 양의 정수 K(≤ N)가 주어진다. 이제 순서대로 K번째 사람을 제거한다. 한 사람이 제거되면 남은 사람들로 이루어진 원을 따라 이 과정을 계속해 나간다. 이 과정은 N명의 사람이 모두 제거될 때까지 계속된다. 원에서 사람들이 제거되는 순서를 (N, K)-요세푸스 순열이라고 한다. 예를 들어 (7, 3)-요세푸스 순열은 <3, 6, 2, 7, 5, 1, 4>이다.
N과 K가 주어지면 (N, K)-요세푸스 순열을 구하는 프로그램을 작성하시오.

입력
첫째 줄에 N과 K가 빈 칸을 사이에 두고 순서대로 주어진다. (1 ≤ K ≤ N ≤ 5,000)

출력
예제와 같이 요세푸스 순열을 출력한다.

n, k = map(int, input().split())
table = [i for i in range(1, n + 1)] # 원으로 앉은 사람들

answer = [] # 제거된 사람들
num = 0 # 제거할 사람의 index

for t in range(n):
    num += k - 1  
    if num >= len(table): # 한 바퀴 돌 때를 대비해 나머지로 선언  
        num = num % len(table)
 
    answer.append(str(table.pop(num)))
print("<", ", ".join(answer)[:], ">", sep='')

해당 문제는 원에 해당하는 데이터 배열 형태를 어떻게 표현해야 할지 생각이 좀 필요했다. 먼저, 제거할 사람들을 찾기 위해 인덱스로 접근하기로 했고, 원(=순환) 형식으로 반복해야 하니 전체 길이로 나누었을 때의 나머지를 떠올렸다. 

제거될 때마다 해당 사람의 인덱스를 answer 리스트에 추가했고 문제에서 요구하는 대로 답을 출력했다. 앞으로 코딩 문제에서 데이터 배열을 순환해야 한다면 전체 길이로 나눈 나머지로 인덱스에 접근하는 것을 생각하는 것이 좋겠다.