[PCCP] 큐(Queue) - 공주 구하기 | 파이썬

코딩테스트 역량 강화 교육(거점형 특화 프로그램)이라는 프로그램에 참여해 공부한 내용입니다.

• IT 직무로 취업을 희망하는 지원자들이 코딩테스트를 통과할 수 있는 알고리즘을 활용한 프로그래밍 교육이며, PCCP 자격증 취득이 목표인 프로그램
• 상세 설명 - 수원대학교(대학일자리 플러스센터)

---

큐(Queue) - 공주 구하기

앞서 공부한 큐을 사용해 공주 구하기 문제를 풀어보겠다.

---

문제

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다.
정보 왕국에는 왕자가 n명이 있는데 서로 공주를 구하러 가겠다고 합니다.
정보왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했습니다.
왕은 왕자들을 나이 순으로 1번부터 n번까지 차례로 번호를 매긴다.
그리고 1번 왕자부터 N번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다.
그리고 1번 왕자부터 시계방향으로 돌아가며 1부터 시작하여 번호를 외치게 한다.
한 왕자가 k(특정숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 된다.
그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다.
이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.

예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자.
처음에는 3번 왕자가 3을 외쳐 제외된다.
이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게된 7번 왕자에게 공주를 구하러갑니다.
n과 k가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.

---

입출력 예

n	k	result
8	3	7
20	3	20
100	5	47

---

제한사항

• N(5<=N<=1,000)
• K(2<=K<=9)

---

코드

from collections import deque
def solution(n, k):

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

    while len(dq) > 1:
        for _ in range(k-1):
        
            # 1 ~ k-1까지 뒤에 다시 삽입
            dq.append(dq.popleft())

        # k번째는 그냥 삭제
        dq.popleft()
                    
    return dq[0]

print(solution(8, 3))
print(solution(20, 3))
print(solution(100, 5))

---

풀이

1. 왕자는 1번부터 n번까지 있으므로, 큐를 range를 이용해 1 ~ n+1 범위로 생성
2. 3번째마다 왕자를 제외시켜야하므로 1, 2번째는 빼서 뒤에 다시 삽입하고 3번째는 그냥 삭제
3. 그렇게 돌다가 dq에 마지막 왕자가 남으면 그 왕자를 return