[코딩테스트] 프로세스 JAVA / Priority Queue (우선순위 큐)

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

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 … 과 같이 표현합니다.

입출력 예
priorities            location return
[2, 1, 3, 2]            2             1
[1, 1, 9, 1, 1, 1]    0              5

입출력 예 설명
예제 #1
문제에 나온 예와 같습니다.

예제 #2
6개의 프로세스 [A, B, C, D, E, F]가 대기 큐에 있고 중요도가 [1, 1, 9, 1, 1, 1] 이므로 [C, D, E, F, A, B] 순으로 실행됩니다. 따라서 A는 5번째로 실행됩니다.

---

이 문제의 핵심은 우선순위 큐다! 

1. 우선순위가 높은 숫자가 먼저 나오게 하는 큐 정의

PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());

2. 배열에 있는 수를 우선순위 큐에 저장

for(int x : priorities) {
	pq.offer(x);
 }

add()	우선순위 큐에 원소를 추가. 큐가 꽉 찬 경우에 에러 발생
offer()	우선순위 큐에 원소를 추가. 값 추가 실패 시 false 반환

 

3. 큐에 값이 없어질 때 까지 반복

이 반복문 안에서 배열 길이만큼 반복문 시작.

 우선순위가 가장 높은, 큐에서 제일 먼저 나오는 숫자가 배열의 숫자와 동일해질 때 

큐에서 해당 숫자를 꺼냄 (poll) . 순번 체크 변수에 1을 더해줌 (answer++) 

만약 location이랑 i 가 같으면 1을 더해줘서 return  (i는 0부터 시작하기 때문에? )

 

4. 배열 반복문이 끝났을 때 큐에 아직 값이 남아있다면 다시 0부터 반복 

---

최종 코드

import java.util.*;

class Solution {
    public int solution(int[] priorities, int location) {
        int answer = 0;
        
        //우선순위가 높은 숫자가 먼저 나오게 하는 우선순위 큐
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        
        for(int x : priorities){
            pq.offer(x);
        }
        
        while(!pq.isEmpty()){
            for(int i=0 ; i < priorities.length; i++){
                if(priorities[i] == pq.peek()){
                    pq.poll();
                    answer++;
                    if(location == i){
                        return answer ++;
                    }
                }
            }
        }
        return answer;
    }
}

 

이 문제는 이해가 좀 어렵다.