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

 

#include <string>
#include <vector>
#include <queue>
using namespace std;

int solution(vector<int> priorities, int location) {
    int answer = 0;
    priority_queue<int> pq;
    queue< pair<int, int> > Q;
    
    for( int i = 0; i<priorities.size(); i++){
        pq.push(priorities[i]);
        Q.push(make_pair(priorities[i], i));
    } 
    
    while(!Q.empty()){
        int value = Q.front().first;
        int index = Q.front().second;
        Q.pop();
        
        if(pq.top() == value){
            pq.pop();
            answer += 1;
            if(index == location){
                break;
            }
        }
        else{
            Q.push(make_pair(value, index));
        }
    }
    
    return answer;
}

 

우선순위에 의해서 큐가 계속 섞이기 때문에 원래의 인덱스 값이 없어진다.

그래서 pair queue Q를 만들어 원래의 중요도 값과 인덱스를 넣는다.

 

여기까지는 나랑 똑같은데 나는 되게 지저분하게 Q 와 pq를 비교했다.

참조만하고 그게 pq랑 같으면 pop 아ㅣ니면 pop해서 push하고 좀 더러웠다.

 

내가 참고한 풀이는 아래와 같다.

1. 일단 Q를 가져와서 중요도 값이랑 인덱스 값을 value, index에 저장한다.

2. Q.pop() 한다. 

 

3-1. 만약 value가 pq.top과 같다면 pq.pop() 한다.

3-2. answer은 프린팅되는 횟수를 카운트한다. 그러니까 인쇄 순서와 같다.

3-3. 그리고 이게 내가 찾던 location과 같은지 index와 비교한다.

3-4. 같다면 index가 몇번째로 출력되었는지 담고있는 answer을 리턴한다.

 

4. 만약 value와 pq.top이 같지 않다면 빼낸 Q를 다시 queue의 끝에 넣는다.