[프로그래머스] 프로세스

https://school.programmers.co.kr/learn/courses/30/lessons/42587

프로그래머스

문제 유형 : 스택 / 큐

문제 난이도 : Level 2

 

문제 설명

운영체제의 역할 중 하나는 컴퓨터 시스템의 자원을 효율적으로 관리하는 것입니다. 이 문제에서는 운영체제가 다음 규칙에 따라 프로세스를 관리할 경우 특정 프로세스가 몇 번째로 실행되는지 알아내면 됩니다.

1. 실행 대기 큐(Queue)에서 대기중인 프로세스 하나를 꺼냅니다.
2. 큐에 대기중인 프로세스 중 우선순위가 더 높은 프로세스가 있다면 방금 꺼낸 프로세스를 다시 큐에 넣습니다.
3. 만약 그런 프로세스가 없다면 방금 꺼낸 프로세스를 실행합니다.
  3.1 한 번 실행한 프로세스는 다시 큐에 넣지 않고 그대로 종료됩니다.

예를 들어 프로세스 4개 [A, B, C, D]가 순서대로 실행 대기 큐에 들어있고, 우선순위가 [2, 1, 3, 2]라면 [C, D, A, B] 순으로 실행하게 됩니다.

현재 실행 대기 큐(Queue)에 있는 프로세스의 중요도가 순서대로 담긴 배열 priorities와, 몇 번째로 실행되는지 알고싶은 프로세스의 위치를 알려주는 location이 매개변수로 주어질 때, 해당 프로세스가 몇 번째로 실행되는지 return 하도록 solution 함수를 작성해주세요.

---

제한사항

• priorities의 길이는 1 이상 100 이하입니다.
  • priorities의 원소는 1 이상 9 이하의 정수입니다.
  • priorities의 원소는 우선순위를 나타내며 숫자가 클 수록 우선순위가 높습니다.
• location은 0 이상 (대기 큐에 있는 프로세스 수 - 1) 이하의 값을 가집니다.
  • priorities의 가장 앞에 있으면 0, 두 번째에 있으면 1 … 과 같이 표현합니다.

 

풀이

큐에 배열을 담고, 내림차순 정렬된 배열도 하나 만들어준다. 이 배열은 남은 요소들 중 가장 큰 수를 가리킵니다.

큐에서 남은요소의 배열 중 가장 큰 값이 올 때까지, 빼서 다시 넣습니다.

이 과정에서 location 요소의 차례일 때, 아직 location보다 더 큰 수가 있다면, location도 그렇게 해주고, 아니라면 location을 빼서 넣습니다.

아니라면, 반복을 그만두고, 실행 횟수를 반환합니다.

코드

C++

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

using namespace std;

int solution(vector<int> priorities, int location) {
    int answer = 0;
    int target = priorities[location];
    
    queue<int> q;
    vector<int> greatest;
    for(int p : priorities) {
        q.push(p);
        greatest.push_back(p);
    }

    sort(greatest.begin(), greatest.end(), greater<int>());
    int idx = 0;
    
    while(!q.empty()) {
        while(q.front() != greatest[idx]) {
            q.push(q.front());
            q.pop();
            if(location == 0) {
                location = q.size();
            }
            location--;
        }
        if(location == 0) {
            answer++;
            break;
        }
        q.pop();
        idx++;
        answer++;
        location--;
    }
    
    return answer;
}