[인프런] 공주 구하기

---

문제설명

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

 

 

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

입력조건

• 첫 줄에 자연수 N(5 <= N <= 1,000)과 K(2 <= K <= 9)가 주어진다.

출력조건

• 첫 줄에 마지막 남은 왕자의 번호를 출력한다.

사고과정

• 데큐를 활용해서 구현했다. 나는 K번 될 때를 cnt라는 변수로 정의해서 +1시켰지만 강의 속 풀이는 K번 loop를 돌도록 구현했다.

풀이

- 나의 풀이

from collections import deque

n, k = map(int, input().split())
cnt = 0
queue = deque([x for x in range(1, n+1)])
while len(queue) != 1:
    cnt += 1
    if cnt == k:
        queue.popleft()
        cnt = 0
    else:
        queue.append(queue.popleft())
print(queue[0])

- 강의 풀이

# 강의 Solution -> k번째니까 k번 loop 도는 것으로 활용!
from collections import deque

n, k = map(int, input().split())
dq = list(range(1, n+1))
dq = deque(dq)

while dq:
    for _ in range(k-1):
        cur = dq.popleft()
        dq.append(cur)
    dq.popleft()
    if len(dq) == 1:
        print(dq[0])
        dq.popleft()  # 이렇게 하면 dq가 비어서 while문 빠져나옴!