[백준] 1966번 프린터 큐

2334 단어 백준백준

[백준] 1966번 프린터 큐

문제 링크: https://www.acmicpc.net/problem/1966

문제 및 입출력

문제 접근

큐 문제이다. 나는 우선 순위 큐와 큐를 같이 사용했다. 우선 순위 큐로는 현재 큐 내에 뽑을 친구를 표시해준다. 큐에는 우선 순위 넘버와 index를 함께 넣어준다.
만약 우선 순위 큐의 top과 큐의 front를 같다면, 큐의 front의 index넘버와 target넘버를 비교를 하고 같다면, 해당 계산을 끝내준다. 만약, 다르다면, 둘다 뽑고, 몇번째 인지 count를 해준다.
우선 순위 큐의 top과 큐의 front와 다르면, 큐의 front를 빼서 끝으로 넣어준다.
시간 복잡도 O(nlogn) 우선 순위를 배치하는 것이 logn이 걸리고, 최악의 경우 n번 반복해야되기 때문에 O(nlogn)이 걸린다.

코드 구현(c++)

#include <iostream>
#include <queue>

using namespace std;

priority_queue<int> pq;
queue<pair<int, int> > q;
int turn, N, target;

int print(){
    int cnt = 1;
    while(!q.empty()){
        if(pq.top() == q.front().second){
            if(target == q.front().first) break;
            pq.pop();
            q.pop();
            cnt++;
        }
        else{
            int first = q.front().first;
            int second = q.front().second;
            q.pop();
            q.push(make_pair(first,second));
        }
    }
    return cnt;
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin >> turn;
    for(int i = 0 ; i < turn ; i++){
        cin >> N >> target;
        int temp;
        for(int j = 0 ; j < N ; j++){
            cin >> temp;
            pq.push(temp);
            q.push(make_pair(j,temp));
        }
        cout << print() << "\n";
        while(!q.empty()) q.pop();
        while(!pq.empty()) pq.pop();
    }

}

좋은 웹페이지 즐겨찾기