[programmers] 스택큐, 프린터

코딩테스트 연습 - 프린터

 

문제

중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했다.

1. 인쇄 대기목록의 가장 앞에 있는 문서(J)를 대기목록에서 꺼냅니다.
2. 나머지 인쇄 대기목록에서 J보다 중요도가 높은 문서가 한 개라도 존재하면 J를 대기목록의 가장 마지막에 넣습니다. 3. 그렇지 않으면 J를 인쇄합니다.

내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 알고 싶습니다.

현재 대기목록에 있는 문서의 중요도가 순서대로 담긴 배열 priorities와 내가 인쇄를 요청한 문서가 현재 대기목록의 어떤 위치에 있는지를 알려주는 location이 매개변수로 주어질 때, 내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 return 하도록 solution 함수를 작성해주세요.

 

해결

key point, 큐 + 우선순위 큐

1. queue가 q, priority_queue가 pq 일때,
2. q에는 문서의 순서를, pq에는 문서의 중요도를 넣는다.
   • priority_queue 는 내림차순으로 자동정렬되는 점을 이용한다.
3. q의 맨 앞에 있는 문서의 중요도와 pq의 맨 앞에 있는 중요도가 같으면 인쇄가 된다. → answer++
   • 이때, 인쇄하는 문서의 순서가 내가 인쇄를 요청한 문서의 번호와 같다면 종료한다.
   • 그렇지 않다면 연산을 계속한다.
4. q와 pq의 맨 앞의 중요도가 다르다면 q맨 앞에 있는 문서를 맨 뒤에 집어넣는다.

 

코드