[프로그래머스] [C++] 프린터 (queue)

1. 문제

코딩테스트 연습 - 프린터 | 프로그래머스 (programmers.co.kr)

코딩테스트 연습 - 프린터

 

2. priority queue

우선순위 큐를 만들고

 

대기목록의 인덱스를 증가시키면서

 

우선순위 큐랑 비교하며

 

top이랑 같지 않으면 대기 목록의 제일 뒤로 보낸다.

 

같으면 우선순위 큐를 pop한다.

 

그리고 만약 location == idx 일 경우 location을 새위치로 업데이트 시켜준다.

 

#include <string>
#include <vector>
#include <queue>
 
using namespace std;
 
int solution(vector<int> priorities, int location) {
    priority_queue<int> pq(priorities.begin(), priorities.end());
    int idx = 0, count = 0;
    while(true) {
        if(idx == location && pq.top() == priorities[location])
            break;
        if(idx == location) {
            location = priorities.size(); 
        }
        if(pq.top() == priorities[idx]) {
            pq.pop();
            count++;
        } 
        else 
            priorities.push_back(priorities[idx]);
        idx++;
    }
        
    return count+1;
}