프로그래머스: 프린터

문제

코딩테스트 연습 - 프린터

 

문제 풀이

 중요도가 가장 높은 문서부터 인쇄될 수 있도록 하여, 내가 요청한 문서는 몇 번째에 인쇄가 되는지 반환하는 문제이다. 이를 찾기 위해 인자로 주어진 위치(location)를 변경하고 중요도가 낮을 경우 큐를 활용해 뒤로 보내는 방식을 사용하면 된다. 이를 통해 내가 출력하고자 하는 문서가 언제 출력되는지 판단할 수 있다.

 

 예를 들어 예제 #1의 경우, [2, 1, 3, 2]이고 출력하고자 하는 문서 2는 가장 먼저 출력된다. 이는 출력 대기열의 최대 값과 인쇄하고자 하는 문서가 일치한다면 횟수를 증가시키는 방식으로 카운트할 수 있다.

 

• priorities를 `deque`를 사용하여 큐로 만든다.
• 현재 출력하고자 하는 문서가 인쇄 대기 중인 문서 중 최대값이라면, 출력한다.
  • 문서가 출력되면, answer를 1 증가 시킨다. 
  • location이 0이라면 내가 출력하고자 하는 문서가 출력된 것이므로 반복문을 종료한다.
  • location이 0이 아니라면, 내가 출력하고자 하는 문서가 아니고 다른 것이 출력된 것이므로 location을 1 감소시킨다.
• 아닌 경우 출력하고자 하는 문서를 인쇄 대기열의 끝으로 보낸다.
  • location이 0 인경우 내가 인쇄하고자 하는 문서의 위치를 현재 인쇄 대기열의 끝으로 변경하여야 한다.
  • location이 0이 아니라면, 다른 문서가 출력되고 앞으로 한 칸 나아가게 되므로 location을 1 감소시킨다.

 

코드

from collections import deque


def solution(priorities, location):
    answer = 0
    max_priority = max(priorities)
    priorities = deque(priorities)

    while True:
        priority = priorities.popleft()
        if max_priority == priority:
            answer += 1
            if not location:
                break
            else:
                location -= 1        
            max_priority = max(priorities)
        else:
            priorities.append(priority)
            location = len(priorities) - 1 if not location else location - 1
    return answer