[C++] 백준 1966 - 프린터 큐(priority_queue)

0. 주어진 문제

프린터 큐

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2 초	128 MB	9834	4930	4078	53.140%

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.

1. 현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.
2. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.

예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때, 어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.

입력

첫 줄에 test case의 수가 주어진다. 각 test case에 대해서 문서의 수 N(100이하)와 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue의 어떤 위치에 있는지를 알려주는 M(0이상 N미만)이 주어진다. 다음줄에 N개 문서의 중요도가 주어지는데, 중요도는 1 이상 9 이하이다. 중요도가 같은 문서가 여러 개 있을 수도 있다. 위의 예는 N=4, M=0(A문서가 궁금하다면), 중요도는 2 1 4 3이 된다.

출력

각 test case에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

1. 풀이

1) priority_queue를 활용해서 정렬을 손쉽게 하였다.

2) 사실 queue를 사용하지 않고 빠르게 하는 방법이 있는 것 같다.

만일, 자신이 찾는 우선순위와 같은 우선순위가 없는 경우에는 자신보다 큰 수를 모두 카운트 하고, 자신을 카운트하고 출력.

자신과 같은 우선순위가 있는 경우라면?

우선 내 우선순위보다 큰 수를 카운트 한 이후, 그중에서 가장 낮은 우선순위의 수를 기준으로 그 수와 자신 사이에 몇개의 같은 우선순위가 있는지를 더해주면 될 것이라 예상된다.

2. 소스코드

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36

#include <iostream> #include <queue>   int main() {     int test, n, m, input;     std::cin >> test;     while(test--) {         int count(0);         std::queue<std::pair<int, int>> queue;         std::priority_queue<int> priority_queue;         std::cin >> n >> m;         for (int i = 0; i < n; i++) {             std::cin >> input;             //내가 넣은 순서를 판단해야하기에 pair로 집어넣었다.             queue.push({ input,i });             //priority_queue를 활용해 우선도에 따라 배열이 자동으로 정렬된다.             priority_queue.push(input);         }         while (queue.size()) {             int priority = queue.front().first;             int num = queue.front().second;             if (priority_queue.top() == priority) {                 count++;                 priority_queue.pop();                 queue.pop();                 //이번 순서가 내가 원하는 순서였다면,                 if (num == m) break;             }             else {                 queue.pop();                 queue.push({ priority, num });             }         }         std::cout << count << "\n";     } } Colored by Color Scripter

cs

3. 문제 출처

https://www.acmicpc.net/problem/1966

4. 참고