[BOJ/백준] 28066번 PYTHON

유승한·2024년 1월 16일
0

algorithm

목록 보기
3/6

문제

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

유형

배운 점

처음에는 queue로 해결하고자 했다.
하지만 pop(0)의 시간복잡도 O(n)이어서 복잡도가 O(num^2 * k)로 시간초과가 발생했다.
따라서 deque를 이용해 해결했다.

코드

from collections import deque
num, k = map(int, input().split())
dq = deque()
for i in range(num):
    dq.append(i+1)

while len(dq) >= k :
    rel = dq.popleft()
    dq.append(rel)
    for i in range(k-1):
        dq.popleft()
print(dq.popleft())

0개의 댓글

관련 채용 정보