Problem : 프린터 큐

유형 : 큐

문제 설명

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

현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다. 예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

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

예제 입력

예제 출력

1
2
5

해결 전략

우선순위 큐를 두고, 각 문서들의 우선순위를 관리한다.
가장 높은 우선순위는 top에 있을 것이다.

문서들을 전부다 큐에 넣고, 가장 맨 앞의 문서를 확인한다.
해당 문서의 우선순위가 우선순위 큐의 top 이라면 해당 문서는 인쇄될 수 있다.

만약 원하는 인쇄 되는 것이 찾는 문서라면 몇 번째로 인쇄 되었는지 알려준다.

주의할 점

궁금한 문서의 인덱스는 0부터 시작이다.

풀이

문서들을 입력받는다.

우선순위 큐에는 인쇄 우선순위를 넣는다.
프린터 큐에는, 문서의 초기위치 인덱스와, 우선순위를 넣는다.

출력이 가능한지 확인한다.

Heap의 top에는 최고 우선순위가 놓여있다.
top과 출력하려는 문서의 우선순위를 비교한다.
- 같다면 인쇄가 가능하다. 이때 출력하는 문서가 궁금했던 문서였는지 index를 확인한다.
- 궁금했던 문서가 아니었다면 출력 횟수를 증가시킨다.
출력이 불가능하다면, 다시 큐에 넣는다.

코드

#include <bits/stdc++.h>

#define pp pair<int, int>
#define index first
#define value second 

using namespace std;

int t, n, want_document;

int priority_check(priority_queue<int> &pq, queue<pp> &printer) {
    int cnt = 1;
    while (!printer.empty()) {
        pp cur = printer.front(); printer.pop();

        if (cur.value == pq.top()){
            pq.pop();
            if (cur.index == want_document)return cnt;
            cnt++;
        }
        else printer.push(cur);
    }
}

void solve() {
    while (t--) {
        priority_queue<int> pq;
        queue<pp> printer;

        cin >> n >> want_document;
        int prior;

        for (int i = 0; i < n; ++i) {
            cin >> prior;
            printer.push({ i,prior });
            pq.push(prior);
        }
        cout << priority_check(pq, printer) << "\n";
    }
}

void input() {
    cin >> t;
}

int main()
{
    ios_base::sync_with_stdio(false);
    //freopen("input.txt", "r", stdin);
    cin.tie(NULL); cout.tie(NULL);

    input();
    solve();

    return 0;
}

피드백

출력할때마다 cnt를 증가 시켜야 한다.