프로그래머스 - 프린터(Stack / Queue) / Level 2

문제 설명

일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다.

이 새롭게 개발한 프린터는 아래와 같은 방식으로 인쇄 작업을 수행합니다.

 

1.  인쇄 대기목록의 가장 앞에 있는 문서(J)를 대기목록에서 꺼냅니다.

 

2.  나머지 인쇄 대기목록에서 J보다 중요도가 높은 문서가 한 개라도 존재하면 J를 대기목록의 가장 마지막에 넣습니다.

 

3.  그렇지 않으면 J를 인쇄합니다.

 

예를 들어, 4개의 문서(A, B, C, D)가 순서대로 인쇄 대기목록에 있고 중요도가 2 1 3 2 라면

C D A B 순으로 인쇄하게 됩니다. 내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 알고 싶습니다.

위의 예에서 C는 1번째로, A는 3번째로 인쇄됩니다.

현재 대기목록에 있는 문서의 중요도가 순서대로 담긴 배열 priorities와 내가 인쇄를 요청한 문서가

현재 대기목록의 어떤 위치에 있는지를 알려주는 location이 매개변수로 주어질 때,

내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 return 하도록 solution 함수를 작성해주세요.

 

제한 사항

• 현재 대기목록에는 1개 이상 100개 이하의 문서가 있습니다.
• 인쇄 작업의 중요도는 1~9로 표현하며 숫자가 클수록 중요하다는 뜻입니다.
• location은 0 이상 (현재 대기목록에 있는 작업 수 - 1) 이하의 값을 가지며 대기목록의 가장 앞에 있으면 0, 두 번째에 있으면 1로 표현합니다.

 

입출력 예시

priorities	location	return
[1, 1, 9, 1, 1, 1]	0	5
[2, 1, 3, 2]	2	1

입출력 예 설명

 

예제 #1

문제에 나온 예와 같습니다.

 

예제 #2

6개의 문서(A, B, C, D, E, F)가 인쇄 대기목록에 있고 중요도가 1 1 9 1 1 1 이므로 C D E F A B 순으로 인쇄합니다.

 

 

문제 풀이

class Document:
    def __init__(self, priority, target):
        self.priority = priority
        self.target = target


def solution(priorities, location):
    answer = 0
    queue = []
    print_count = 0

    for i in range(len(priorities)):  # queue를 생성하여 Document 객체를 초기화 시켜서 넣어준다.
        doc = Document(priorities[i], False)
        if i == location:
            doc.target = True
        queue.append(doc)

    while True:  # target이 인쇄 될때까지 하나씩 queue에서 꺼내서 우선순위가 제일 높으면 인쇄하고 아니면 다시 queue에 넣는다.
        doc = queue.pop(0)
        max_val = 0

        for i in queue:  # queue에 남아있는 것들 중 가장 높은 우선순위를 찾는다.
            if i.priority > max_val:
                max_val = i.priority

        if max_val > doc.priority:  # 보류하고 다시 queue에 넣는다.
            queue.append(doc)
        else:  # 인쇄한다.
            print_count += 1
            if doc.target:
                break

    answer = print_count
    return answer

작업순서를 유지한 채로 queue를 구성해야 되는 문제이다.

그런 다음 queue에서 하나씩 꺼내어 우선순위에 따라 다시 집어넣거나 인쇄를 실행하거나 하면 된다.

queue에 넣는 객체는 우선순위에 대한 정보와 내가 요청한 인쇄인지에 대한 정보를 담고 있어야 되기 때문에 

Document class를 정의하여 Document 객체에 정보를 담았다.

이 방법은 코드가 더 복잡해질 수 있지만 보다 논리적으로 좀 더 완벽하게 구현하고자 class를 정의하였다.

 

 

이 문제가 왜 Stack/Queue 카테고리에 속해있을까?

인쇄 대기목록의 작업순서를 유지해야 되기 때문에 Queue 자료구조를 썼다.

우선순위와 관계없이 인쇄 대기목록에서 순서대로 하나씩 꺼내어 규칙에 따라 판단해야 되기 때문이다.