문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/42587
문제 설명
운영체제의 역할 중 하나는 컴퓨터 시스템의 자원을 효율적으로 관리하는 것입니다. 이 문제에서는 운영체제가 다음 규칙에 따라 프로세스를 관리할 경우 특정 프로세스가 몇 번째로 실행되는지 알아내면 됩니다.
- 실행 대기 큐(Queue)에서 대기중인 프로세스 하나를 꺼냅니다.
- 큐에 대기중인 프로세스 중 우선순위가 더 높은 프로세스가 있다면 방금 꺼낸 프로세스를 다시 큐에 넣습니다.
- 만약 그런 프로세스가 없다면 방금 꺼낸 프로세스를 실행합니다.
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 … 과 같이 표현합니다.
입출력 예
정답 코드
import java.util.Collections;
import java.util.PriorityQueue;
class Solution {
public int solution(int[] priorities, int location) {
int count = 0;
PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
for (int priority : priorities) {
pq.add(priority);
}
LOOP:
while (!pq.isEmpty()) {
for (int i = 0; i < priorities.length; i++) {
if (pq.peek() == priorities[i]) {
pq.poll();
count++;
if (location == i) {
break LOOP;
}
}
}
}
return count;
}
}
값이 큰 값부터 나오는 큐를 만들기 위해 우선순위 큐인 pq를 선언한 후 Collections.reverseOrder()를 통해 우선순위 큐의 기본구현체인 mini heap에서 max heap으로 바꾸어준다.
그 후 priorities 배열을 for-each문으로 돌려 값을 하나씩 작은 값이 순서대로 나오는 우선순위 큐인 pq에 넣어준다.
이후 pq가 완전히 비워지기 전까지나 목표 타켓인 priorities[location]을 찾기 전까지 반복하는 while문을 정의하고 그 아래에 priorities 배열을 for문으로 하나씩 꺼내와 pq.peek()로 현재 가장 큰 수가 맞으면 꺼내고 꺼낸 후 목표인 priorities[location]을 몇 번째에 꺼내는지 기록하기 위해 count을 1씩 증가시킨다.
만약 꺼내진 원소의 인덱스가 location이라면 priorities[location]을 꺼낸다는 목표를 완료하였기에 반복문을 탈출하고 몇 번째로 꺼내지는 것인지 기록한 count을 정답으로 반환하면 된다.
