[c++] 프로그래머스 - 프린터(Level 2)

문제

코딩테스트 연습 - 프린터

 

코드

// 01:19(문제 이해 시작)
// 01:27(문제 구현 시작) ~ 01:44(풀이 완료)
#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <algorithm>
#include <queue>

using namespace std;

int solution(vector<int> priorities, int location) {
    
    int answer = 0;
    map<int,int>docs;
    queue<int>q;
    
    
    for(int i=0; i<priorities.size(); i++){
        docs[i] = priorities[i]; // 문서마다 우선순위 부여
        q.push(i);  //문서 이름 부여
    }
    
    
    sort(priorities.begin(), priorities.end(), greater<>());
    
    
    while(!priorities.empty()){
        if(docs[q.front()]!=priorities.front()){  //현재 뺴넨 문서보다 우선순위 문서가 존재한다면
            int doc = q.front();
            q.pop();
            q.push(doc);
            
        }
        else{
            answer++;
            priorities.erase(priorities.begin());
            if(q.front()==location) break;
            q.pop();
        }
    }
    
    
    return answer;
}

 

풀이(17분)

먼저 docs라는 map을 통해 각 문서의 key값은 정수로, value값은 우선순위 값으로 넣어줬습니다. 그다음 priorities 배열은 내림차순으로 정렬한 뒤 현재의 문서가 priorities 배열의 첫 번째 값과 같다면 인쇄 횟수를 추가해주고, 찾는 문서가 현재라면 바로 출력해주었습니다. 만약 현재 문서가 priorities 배열의 첫번째 값과 다르다면 우선순위가 더 큰 문서가 있다는 말이 되기에 문제의 조건대로 대기열의 젤 마지막으로 보내버리는 방식으로 구현했습니다.