요세푸스 문제 0

문제링크

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

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)

 

출력

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

 

순열이 원에서 빼진 수가 첫 수로 돌아가서 많이 고민이 된 문제이다!

n, k = list(map(int, input().split()))

nums = list(range(1, n + 1))

result = []
idx = 0

while nums:
    idx = (idx + k - 1) % len(nums)
    result.append(nums.pop(idx))

print(str(result).replace('[', '<').replace(']', '>'))

인덱스를 뺀 수 의 인덱스로 초기화 해주고...

배열의 숫자로 나눈 나머지를 다시 인덱스로 해주는 코드이다!

배열의 순서가 고정되어 있어 인덱스가 헷갈린다...

또, 출력을 할때 리스트의 괄호가 <, >라서 문자열로 바꿔준 후 [,]를 각각 <. >로 바꿔줬다!

 

이번에는 deque를 이용해서 배열의 순서를 바꿔주면서 풀어보자!(지민이형 코드 참고)

from collections import deque

n, k = list(map(int, input().split()))

deq = deque(range(1, n + 1))

result = []

while deq:
    deq.rotate(-(k-1))
    result.append(deq.popleft())

print(str(result).replace('[', '<').replace(']', '>'))

일단 배열이 주어지면

k-1번 만큼 왼쪽으로 이동해서...

popleft()로 첫 수(원래 배열에서 k번째 수)를 빼준다!

deque가 빌때 까지 반복한다...

 

이 풀이가 좀 더 직관적이고 이해하기 쉬운 것 같다!