[프로그래머스] 운영체제 (PCCP 모의고사 #1 4번)

https://school.programmers.co.kr/learn/courses/15008/lessons/121686

프로그래머스

프로그래머스 - 운영체제

문제 유형 : 우선순위 큐, 큐

문제 난이도 : Level 3(개인적인 체감)

 

문제 설명

개발자 준모는 운영체제를 만들었습니다. 준모가 만든 운영체제는 프로그램의 우선순위와 호출된 시각에 따라 실행 순서를 결정합니다. 모든 프로그램에는 1부터 10까지의 점수가 매겨져 있으며, 이 점수가 낮을수록 우선순위가 높은 프로그램입니다. 각 프로그램들은 실행 시간이 정해져 있으며 프로그램이 호출되면 대기상태에 있다가 자신의 순서가 되면 실행 시간 동안 실행된 뒤 종료됩니다.

준모가 만든 운영체제는 호출된 프로그램들 중 우선순위가 가장 높은 프로그램을 먼저 실행합니다. 호출된 각 프로그램은 자신보다 우선순위가 높은 호출된 프로그램이 모두 종료된 후에 실행됩니다. 단, 실행 중인 프로그램보다 우선순위가 높은 프로그램이 호출되어도 실행 중이던 프로그램은 중단되지 않고 종료될 때까지 계속 실행됩니다. 또한, 우선순위가 같은 프로그램들 중에서는 먼저 호출된 프로그램이 먼저 실행됩니다.

다음은 1번부터 4번까지의 4개의 프로그램이 호출된 예시입니다.

예를 들어, 1번부터 4번까지 4개의 프로그램의 점수가 순서대로 2, 1, 3, 3이며, 호출된 시각은 0, 5, 5, 12초이고, 수행시간은 10, 5, 3, 2라고 가정해 봅시다.

1. 1번 프로그램이 0초에 호출될 때 실행 중인 프로그램이 없으므로, 0초에 1번 프로그램이 바로 실행됩니다. 1번 프로그램은 10초에 종료되며, 2, 3번 프로그램이 새로 호출됐습니다.
2. 호출된 2, 3번 프로그램 중 2번 프로그램의 점수가 1로 우선순위가 높습니다. 2번 프로그램은 5초에 호출되어 10초에 실행될 때까지 5초 동안 대기했습니다. 2번 프로그램은 15초에 종료되며, 4번 프로그램이 새로 호출됐습니다.
3. 호출된 3, 4번 프로그램은 점수가 같지만, 3번 프로그램이 먼저 호출되었기 때문에 3번 프로그램이 먼저 실행됩니다. 3번 프로그램은 5초에 호출되어 15초에 실행될 때까지 10초 동안 대기했습니다. 3번 프로그램은 18초에 종료됩니다.
4. 4번 프로그램이 마지막으로 실행되며, 4번 프로그램은 12초에 호출되어 18초에 실행될 때까지 6초 동안 대기했습니다. 4번 프로그램은 20초에 종료됩니다.

모든 프로그램이 종료되는 시각은 20초이며, 각 프로그램이 대기한 시간은 순서대로 0, 5, 10, 6초입니다. 점수가 1인 프로그램의 대기시간 합은 5고, 점수가 3인 프로그램의 대기시간 합은 16 임을 알 수 있습니다.

프로그램들의 정보를 나타내는 2차원 정수 배열 program이 주어질 때, 모든 프로그램들이 종료되는 시각과 프로그램의 점수마다 대기시간의 합을 정수 배열에 담아 return 하는 solution 함수를 완성하세요. return 해야 하는 answer 배열은 길이가 11인 정수 배열입니다. answer[0]은 모든 프로그램들이 종료되는 시각을 의미하며, answer[i](1 ≤ i ≤ 10)는 프로그램의 점수가 i인 프로그램들의 대기시간의 합을 의미합니다.

 

 

풀이

작업 우선순위가 낮을수록 먼저 실행시키도록, 우선순위 큐를 만들어준다.

입력받는 작업들도 정렬해준 뒤, 큐에 넣는다. 정렬 기준은 호출순으로 해준다.

 

현재 시간이 만족되는 작업들을 큐에서 빼서 우선순위 큐에 넣는다.

우선순위 큐에 작업이 있다면, 작업을 수행한다.

answer[a]에는 time-b를 누적시킨다. 현재 시간에서 호출 시간을 뺀 값이 대기시간이기 때문이다.

걸린 시간 누적에 작업시간을 더해준다. (time += c)

 

마지막으로 총 걸린 시간을 answer[0]에 넣고 마무리한다.

 

코드

C++

 

#include <bits/stdc++.h>

using namespace std;


struct comp
{
    bool operator()(const vector<int> &a, const vector<int> &b) const
    {
        if(a[0] == b[0]) return a[1] > b[1];
        return a[0] > b[0];
    }
};

vector<long long> solution(vector<vector<int>> program)
{
    priority_queue<vector<int>, vector<vector<int>>, comp> pq;
    const int n = program.size();
    vector<long long> answer(11, 0);
    sort(program.begin(), program.end(), [](const auto &a, const auto &b)
         { 
             if(a[1] == b[1]) return a[0] < b[0];
             return a[1] < b[1];
         });

    queue<vector<int>> q;
    for (const auto &pg : program)
    {
        q.push(pg);
    }

    int time = 0;
    while(!pq.empty() || !q.empty()) {
        while(!q.empty() && q.front()[1] <= time) {
            pq.push(q.front());
            q.pop();
        }
        if(!pq.empty()) {
            auto curr = pq.top();
            pq.pop();
            
            int a = curr[0];
            int b = curr[1];
            int c = curr[2];
            //cout << a << " " << b << " " << c << endl;
            answer[a] += time-b;
            time += c-1;
        }
        time++;
    }
    answer[0] = time;

    return answer;
}