[백준] 우선순위큐 - 1966. 프린터큐

www.acmicpc.net/problem/1966

1966번: 프린터 큐

 

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 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 복사

3

1 0 

5

4 2

1 2 3 4

6 0

1 1 9 1 1 1

예제 출력 1 복사

1

2

5

 

풀이

우선순위큐를 사용할 생각을 못하고 처음에 map을 이용해서 풀려고 했다.

우선순위큐 문제를 자주 접하지 못하다보니 자꾸 까먹는 듯 싶다.

 

우선순위큐를 사용하게 되면 원소는 내림차순 정렬로 기본정렬된다.

priority_queue<int, vector<int> > pq; //자료형 int, 컨테이너 vector, 기본값은 내림차순

 

큐와 우선순위큐를 사용해서 코드를 작성했는데, 예를 들어

 

3 0

1 2 5

 

라고 주어진다면

 

q   : 0, 1 / 1, 2 /  2, 5

pq :  5 / 2 / 1

 

이렇게 원소가 들어가지고 while문을 돌면서 q의 원소를 하나씩 꺼내서 pq의 top이랑 비교하고 같으면 pq에서 pop한다. 이때, index의 값이 M과 같으면 찾으려는 위치이므로 출력해주고 break문으로 빠져나간다. 만약, 같지 않다면 다시 q에 넣어준다.

 

 

//1966. 프린터큐
#include <iostream>
#include <algorithm>
#include <queue>
 
using namespace std;
 
// 오름차순 정렬하고 싶을때(greater 사용도 가능)
// struct compare{
//     bool operator()(int a, int b){
//         return a < b;
//     }
// }
int main(){
    int test_case;
    cin >> test_case;
    vector<int> v;
    for(int i=0; i<test_case; i++){
        int N, M;
        cin >> N >> M;
 
        queue<pair<int, int> >  q;
        priority_queue<int, vector<int> > pq; //자료형 int, 컨테이너 vector, 기본값은 내림차순
 
        for(int j=0; j<N; j++){
            int element;
            cin >> element;
            q.push(make_pair(j, element));
            pq.push(element);
        }
 
        int cnt = 0;
        while(!q.empty()){
            int index = q.front().first;
            int value = q.front().second;
            q.pop();
            if(pq.top() == value){
                pq.pop();
                cnt++;
                if(index == M){
                    cout << cnt << "\n";
                    break;
                }
            }else q.push(make_pair(index, value));
        }
    }
 
    return 0;
}