[백준] 1158 - 요세푸스 문제 (python)

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

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)

출력

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

예제 입력 1 

7 3

예제 출력 1 

<3, 6, 2, 7, 5, 1, 4>

 

 

풀이

from collections import deque
a, b = map(int,input().split())
c = deque([i for i in range(1,a+1)])
d = []
for i in range(a):
    for j in range(1,b+1):
        if j == b :
            d.append(c[0])
            c.popleft()
        else :
            c.append(c.popleft())
print("<",end='')
for i in range(a-1):
    print(d[i],end=', ')
print(d[a-1],end='>')

이렇게 원으로 되어있다고 가정할 때 K=3인 경우엔 1,2,3 / 4,5,6 / 7,1,2 / 4,5,7 / 1,4,5 / 1,4,1 / 4,4,4 의 형식 이므로

c라는 하나의 덱을 만들어서 요소를 집어넣은 후 , 개수 만큼 반복을 하여 그중 반복문에서 b라는 번째가 저장된 변수와 같으면,그 수는 정답 리스트에 추가하고 해당 요소는 삭제하는 식으로 구현하였고, 아닐 경우엔 덱의 뒤에 요소를 추가해주는 형식.

 

 

자료구조 입문한지 얼마 안되어서 그런지 아직 어렵고 시간 초과와 같은 문제가 자주 발생하는데 좀 더 열심히 공부해야겠습니다...