[알각코] 인프런 5-5 공주 구하기

공주 구하기

문제 정보는 인프런의 파이썬 알고리즘 문제풀이 코딩테스트에 있습니다!

문제풀이

이 문제는 큐 자료구조로 해결 합니다.

import sys
from collection import deque
sys.stdin = open("input.txt", "rt")

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()