[프로그래머스/Python] 프린터(큐)

코딩테스트 연습 - 프린터

 

idx를 가지고 다니는 방식으로 작성했다.

 

from collections import deque
def solution(priorities, location):
    answer = 0
    q = deque()
    for idx, paper in enumerate(priorities):
        q.append([paper, idx])
    
    while q:
        cur = q.popleft()
        paper, idx = cur[0], cur[1]

        #우선순위 체크
        flag = 1
        for i in q:
            if i[0] > paper:
                flag = 0
                break
                
        if flag == 0: # 자기보다 큰게 있으면 뒤로 보내기
            q.append([paper, idx])
        if flag == 1: #자기보다 큰게 없으면 print
            answer += 1
            if idx == location:
                break

    return answer