백준 (boj) 파이썬 - 1158 요세푸스 문제

문제 링크 : https://www.acmicpc.net/problem/1158

1158번: 요세푸스 문제

 

 

 

 

N명의 사람들이 있고, K를 주기로 사람들을 제거하는 문제였다. 이때 K(주기)가 사람의 인원수를 초과하면

사람의 인원수로 나눈 나머지로 값을 초기화해주면 된다. 

 

 

 

 

 

파이썬 코드

N,K = map(int,input().split())
arr = [i for i in range(1,N+1)]    # 맨 처음에 원에 앉아있는 사람들

answer = []   # 제거된 사람들을 넣을 배열
num = 0  # 제거될 사람의 인덱스 번호

for t in range(N):
    num += K-1  
    if num >= len(arr):   # 한바퀴를 돌고 그다음으로 돌아올때를 대비해 값을 나머지로 바꿈  
        num = num%len(arr)
 
    answer.append(str(arr.pop(num)))
print("<",", ".join(answer)[:],">", sep='')