[Python] 백준 - 11866 요세푸스 문제 0

문제

요세푸스 문제는 다음과 같다.

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 ≤ 1,000)

출력

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

 

 

원 테이블에서 k번째 순서마다 원테이블을 이탈하는 게임이다.

 

생각해볼 점은 k번째를 어떻게 셀 수 있으며, 원형 테이블처럼 1~7 후 다시 1로 돌아오는 구조를 집적해야 한다는 거다.

 

1. k번째를 세는 것은 for문으로 for i in range(1, k+1)로 i로 돌린다. i == k일 때 조건문 진입한다.

2. 원형 테이블은 while문으로 다른 변수로 구분해서 진행하며 for문이 돌아갈 때마다, 원형 테이블을 돌게 된다.

3. for문은 상위 while문이 멈추는 조건이 되지 않는 이상 계속 돌아야 하며, while문은 테이블에 아무도 없으면 종료된다.

 

- 코드

import sys
import math
sys.setrecursionlimit(10**6)
input = sys.stdin.readline

n, k = map(int, input().split())
d = [i for i in range(1, n+1)]
ans = []

idx = 0
while True:
	if not d:
		break
	for i in range(1, k+1): # 1 2 3
		if idx == len(d):
			idx = 1
		else:
			idx += 1
		if i == k:
			ans.append(d[idx-1])
			d.remove(d[idx-1])
			idx -= 1

string = '<'
for i in range(len(ans)):
	if i != len(ans)-1:
		string += str(ans[i]) + ', '
	else:
		string += str(ans[i]) + '>'
print(string)