(C++) - 프로그래머스(고득점 kit - 스택/큐) : 프린터 답

programmers.co.kr/learn/courses/30/lessons/42587

코딩테스트 연습 - 프린터

우선순위 큐와 큐를 사용해 푼 문제였습니다.

 

풀이방법

 1. 우선순위 큐와 큐에 우선순위 정보를 push합니다.

 2. 우선순위가 같다면 프린트 하고, 문서번호까지 같으면 정답입니다.

Code

#include <bits/stdc++.h>
using namespace std;
 
int solution(vector<int> priorities, int location) {
    int answer = 0, cnt = 0;
    queue<pair<int, int>> q;
    priority_queue <int> pq;
 
    for (int i = 0; i < priorities.size(); i++) {
        q.push({i, priorities[i]});
        pq.push(priorities[i]);
    }
    while (!q.empty()) {
        int index = q.front().first;
        int value = q.front().second;
        q.pop();
        if (pq.top() == value) {
            pq.pop();
            cnt++;
            if (index == location) {
                answer = cnt;
                break;
            }
            continue;
        }
        q.push({index, value});
    }
    return answer;
}