[백준/파이썬] 1158번

백준 1158번

문제

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

코드

from collections import deque

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

people = deque()
for i in range(1, n+1): 
   people.append(i)
result = []

while people:
  for _ in range(k-1):
    people.append(people.popleft())

  result.append(people.popleft())

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

풀이

큐를 이용하여 푸는 문제입니다.
deque를 이용합니다.
k-1번만큼 for문을 돌려 맨 아래의 값을 지워준 후, 다시 맨 위에 추가해줍니다.
이후 맨 아래의 값을 지우고 result에 값을 추가해줍니다.