문제 📚
어느 왕국의 이웃 나라 외동딸 공주가 숲 속의 괴물에게 잡혀갔습니다.
왕국에는 왕자가 N명이 있는데 서로 공주를 구하러 가겠다고 합니다. 왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했습니다.
왕은 왕자들을 나이 순으로 1번부터 N번까지 차례로 번호를 매기고 1번 왕자부터 N번 왕자까지 순서대로 시계 방향으로 돌아가게 동그랗게 앉게 합니다. 그리고 1번 왕자부터 시계 방향으로 돌아가며 1부터 시작하여 번호를 외치게 합니다. 한 왕자가 K(특정 숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 됩니다. 그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외칩니다.
이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있습니다.
input
8 3
output
7
문제 설명 👩🏫
k=3번째 수만 빼준다.
1,2,3,4,5,6,7,8 - 뺀 수
1,2,4,5,6,7,8 - 3
1,2,4,5,7,8 - 6
2,4,5,7,8 - 1
2,4,7,8 - 5
4,7,8 - 2
4,7 - 8
7 - 4
코드 👩💻
from collections import deque
def solution(n, k):
dq = list(range(1, n+1))
prince = deque(dq)
while prince:
for i in range(k-1):
cur = prince.popleft()
prince.append(cur)
prince.popleft()
if len(prince) == 1:
return prince[0]
n, k = map(int, input().split())
print(solution(n,k))양방향에서 더하고 뺄 수 있는 deque 모듈을 이용해서 풀었다. k번째 전까지는 남아있어야 하는 원소이므로 왼쪽의 원소를 빼고 다시 오른쪽에 이어붙인다. k번째 수는 문제에서 설명했던 대로 그냥 빼준다.
이 작업은 deque가 비어있지 않는 동안 계속 반복되고, deque의 길이가 1이 될때 함수 결과값을 return 해줌으로써 끝이 난다.
AI researcher