백준: 1966 프린터

문제

1966번: 프린터 큐

 

문제 풀이

 프린터를 하기 위한 문서 대기열에 문서들이 있다. 이때, 각 문서들은 `우선순위`를 가진다. 큐(문서 대기열)에서 문서를 인쇄하고자 할 때 큐에 있는 문서들 중 우선수위가 하나라도 높은 것이 있다면, 프린트하지 않고 큐의 맨 뒤로 보낸다.

 

 이는 앞서 다룬 프로그래머스: 프린터와 유사한 문제이다. 문제를 풀기 위해서 다음과 같이 접근하였다.

• 최초의 문서가 위치한 인덱스를 알기 위해 `enumerate`를 활용한다.
• `deque`를 활용하여, 인쇄 가능한 상태이면 카운트를 증가하고 아닌 경우 `append`한다.

 

코드

from sys import stdin
from collections import deque


if __name__ == "__main__":
    T = int(stdin.readline())

    for _ in range(T):
        N, M = map(int, stdin.readline().split())
        q = deque(enumerate(map(int, stdin.readline().split())))
        cnt = 0

        while q:
            cur_max = max(q, key=lambda x: x[1])[1]
            idx, priority = q.popleft()

            if priority >= cur_max:
                cnt += 1
                if idx == M:
                    break
            else:
                q.append((idx, priority))

        print(cnt)