프로그래머스 (프린터, 스택/큐) C++

 

map의 개념으로 index와 value가 모두 필요한 문제였다. Map을 이용해서 Pair를 만들고 이를 Queue에 넣어주는 방식으로 접근을 했다가 잘 되지 않았다. 그래서 queue 자체를 한 쌍을 만들어서 할당을 했더니 잘 풀렸다. Queue를 사용해서 한 방에 문제가 풀렸다... 너무 기분이 좋은데...?

 

#include <string>
#include <queue>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

int solution(vector<int> priorities, int location) {
    int answer = 0;
    int index = 0;
    queue<pair<int,int>> q;
    
    // key에 index 값을 넣고 value에 우선순위를 넣어준다.
    for(int i =0 ; i < priorities.size() ; i++) {
        q.push(make_pair(i,priorities[i]));
    }
    
    // 우선순위를 내림차순으로 정렬
    sort(priorities.begin(), priorities.end(), greater<int>());
    
    while(true) {
        // queue의 가장 앞의 원소가 가장 우선순위가 높은 것과 동일한 경우
        if(q.front().second == priorities[index]) {
            answer++;
            index++;
            if(q.front().first == location)
                break;
            q.pop();
        } 
        // 같지 않은 경우 가장 앞의 원소를 제일 뒤로 넣어준다.
        else {
            int first = q.front().first;
            int second = q.front().second;
            q.pop();
            q.push(make_pair(first, second));
        }
    }
    
    return answer;
}