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

1158

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

1158번: 요세푸스 문제

[정답]

import sys
N, K = map(int, sys.stdin.readline().split())
 
a = [i for i in range(1,N+1)]
b = []
c = 0
while a:
  c += K-1
  if c >= len(a):
    c %= len(a)
  b.append(str(a.pop(c)))
print('<', ", ".join(b), '>', sep = '')

.

.

.

[풀이]

배열 a에서 c번(index)마다 항목을 삭제(pop)해준다.

pop해서 반환된 값을 b에 추가해준다.

 

이 때 배열 a에서 값들이 하나씩 삭제되므로

c는 K-1만큼만 증가시켜준다.

 

c가 배열 a의 길이보다 길어지면

처음으로 돌아가 카운트하는 순환 구조이므로

나머지 연산을 이용하여 c를 줄여준다.

 

p.s.

index를 계산하는 방법을 사용해야한다.

배열 자체를 pop과 append를 통해서 뗐다 붙였다 풀었더니

시간초과로 실패했다.