[프로그래머스/C++] 프로세스 (스택,큐 - L2)

자료구조를 활용한 완전 탐색 문제다.

단, 우선순위를 고려해야하는 문제가 있어서 처음에는 우선순위 큐 자료구조를 활용해야하나? 싶었는데, 

막상 원하는 location의 값이 언제 나오는지를 확인하기 위해서는 원래 위치를 기반으로 탐색을 할 수 밖에 없었다.

그래서 location에서 지정해둔 index를 따라가면서 순서를 찾는 방식으로 개발했다.

 

 

다른 방법으로도 풀 수 있을 것 같지만 가장 직관적이게 다가온 방법은 이 방법이었다.

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

using namespace std;

int solution(vector<int> priorities, int location) {
    
    int answer = 0;
    queue<int> q;
    
    // queue 자료구조
    for(int i = 0; i < priorities.size(); i++) q.push(priorities[i]);
    
    //우선순위를 위해서 정렬 후, 해당 숫자가 pop할 수 있도록 작업
    sort(priorities.begin(), priorities.end());
    
    int Aidx = location;
    
    while(true){
        
        //맨 앞 친구가 priority에 맞춰 나올 수 있는지
        //현재 우선순위와 동일한지, 동일 하면 pop
        if(q.front() == priorities.back()){
        
            answer++; 
            
            //근데 이번에 나오는 인덱스가 첫번째 인덱스였을 경우
            //(포인터 개념으로 계속 따라가게끔 할거임)
            if(Aidx == 0) break;
            
            priorities.pop_back(); //우선순위 함수 pop
            q.pop();             //queue 앞도 pop
            
            // 아닐 경우에는 pop만 진행함 <- 필요 없을 수도 있음
            Aidx--; if(Aidx < 0) Aidx = q.size()-1;
        }
        
        //우선 순위가 동일하지 않는다면 뒤로 다시 넣기
        else{
            int temp = q.front();
            q.pop();
            q.push(temp);
            
            // 아닐 경우에는 pop만 진행함 <- 필요 없을 수도 있음
            Aidx--; if(Aidx < 0) Aidx = q.size()-1;
        }
    }
    
    return answer;
}

 

시간 결과