[프로그래머스] 프린터

문제 링크 >> https://programmers.co.kr/learn/courses/30/lessons/42587

 

 

📋 문제

일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다.

그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다.

이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다.

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

 

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로 표현합니다.

 

 

 

💡 사용된 개념

✔️ Queue

FIFO(First-In, First-Out)의 특성을 가지고 순서가 있는 자료구조이다.

data를 꺼낼 때 가장 먼저 들어가있던 순서대로 data를 꺼낼 수 있다.

한쪽 끝에서는 제거, 한쪽 끝에서는 추가가 가능한 터널 형태로 이해하면 쉽다.

 

파이썬에서 Queue는 [] list로 구현할 수 있다.

.pop(0)으로 첫 번째 원소를 꺼낼 수 있다.

.append()로 원소를 추가할 수 있다.

 

 

📝 풀이

def solution(priorities, location):
    answer = 0
    priorities = list(enumerate(priorities))
    
    while priorities:
        target = priorities.pop(0)
        if len(priorities) != 0:
            if target[1] < max(priorities, key=lambda x: x[1])[1]:
                priorities.append(target)
            else:
                answer += 1
                if target[0] == location:
                    break
        else:
            answer += 1
    return answer

 

먼저 입력으로 들어온 priorities를 enumerate()를 이용해 (인덱스, 중요도)로 이루어진 리스트로 만들었다.

priorities를 queue로 삼고 이를 순회하며 중요도를 비교할 예정이다.

 

while문으로 priorities를 순회하며 첫 번째 원소를 꺼내 이를 target에 저장한다.

이때 먼저 priorities에서 원소를 빼내고 있기 때문에 priorities가 빈 리스트인지 확인하지 않으면 이후 max()에서 빈 리스트가 들어가게 되어 에러가 난다.

 

따라서 빈 리스트인지 확인해보고 빈 리스트가 아니라면 target의 중요도와 priorities의 제일 높은 중요도를 비교한다.

target의 중요도보다 높은 것이 priorities에 존재한다면 target은 출력될 수 없으므로 다시 priorities에 넣는다.

target의 중요도보다 높은 것이 priorities에 존재하지 않는다면 target의 중요도가 제일 높다는 것이다.

target은 출력되어야 할 문서이므로 target을 출력한다는 뜻으로 answer에 1을 더한다.

이후 target이 우리가 출력하기를 원하던 문서인지 확인한 후 그 문서가 맞다면 break로 while문을 끝낸다.

 

만약 priorities가 빈 배열이라면 target이 우리가 원하던 문서와 같으며 제일 마지막 순서로 출력되는 문서이다.

따라서 이 경우에는 이 문서를 출력한다는 의미로 단순히 answer에 1을 더해준다.