[Python/Programmers/Level 2] 프린터

출처: https://programmers.co.kr/learn/courses/30/lessons/42587

코딩테스트 연습 - 프린터

* 이 포스팅은 Programmers 코딩 테스트 연습 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
[2, 1, 3, 2]	2	1
[1, 1, 9, 1, 1, 1]	0	5

# 코드 설명

def solution(priorities, location):
    answer = 0
    
    queue = [(i, p) for i, p in enumerate(priorities)]
    
    while priorities:
        value = priorities[0]
        
        if value == max(priorities):
            priorities.pop(0)
            idx = queue.pop(0)
            answer += 1
            
            if idx[0] == location:
                break
            
        else:
            priorities.append(priorities.pop(0))
            queue.append(queue.pop(0))
    
    return answer

 

• answer : 프린터에서 출력되는 횟수를 저장하기 위한 변수입니다.

• queue : 입력으로 전달되는 priorities의 중요도와 location을 저장하는 리스트입니다.

  • enumerate(x) : 입력으로 전달되는 리스트(x)의 인덱스와 값을 함께 반환해주는 함수입니다.

  • value = priorities[0] : 대기 목록 맨 앞 문서의 중요도를 저장합니다.

• if value == max(prioties)

  • 맨 앞 문서의 중요도가 전체 인쇄 대기 목록에서 가장 클 경우

  • 해당 문서를 인쇄하고(priorities.pop, queue.pop)

  • 출력 횟수(answer)를 1 증가시킵니다.

• if idx[0] == location: break

  • 해당 문서가 찾고자 하는 문서일 경우 반복문을 종료합니다.

• else:

  • 맨 앞 문서의 중요도가 전체 인쇄 대기 목록에서 가장 크지 않을 경우

  • 해당 문서를 인쇄 대기 목록의 맨 뒤에 위치시킵니다.