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

 

문제 링크

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

문제

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

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

 

---

 

제출 코드

처음에는 우선순위 큐 하나로 풀이했는데, 그 방법이 순서를 고려하지 않고 우선순위만 순번으로 계산해 오답으로 나왔다.

따라서 실제 인쇄 순서를 나타낸 Queue\와 현재 남은 문서 중 가장 높은 우선순위를 나타내는 PriorityQueue로 구분해 구현하였다.

 

1. queue를 하나 꺼내어 현재 가장 높은 우선 순위를 가졌는지 확인한다.
2. 우선순위가 가장 높다면, 우선순위를 꺼내고 실행하고 실행횟수(count)를 +1 한다.
3. 만약 현재 인덱스가 location과 같다면 실행 횟수를 리턴한다.
4. 우선 순위가 높은 다른 노드가 있다면, queue의 가장 뒤로 보낸다.

 

import java.util.*;
class Solution {
    public int solution(int[] priorities, int location) {

        Queue<Node> queue = new LinkedList<>();
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());

        for (int i = 0; i < priorities.length; i++) {
            queue.offer(new Node(priorities[i], i));
            pq.offer(priorities[i]);
        }

        int ans = 0;
        while (!queue.isEmpty()) {
            Node curr = queue.poll();

            // 현재가 가장 높은 우선순위 인지
            if (curr.prior == pq.peek()) {
                pq.poll();
                ans++;

                if (curr.idx == location) {
                    return ans;
                }
            } else { // 높은 우선순위가 있을 때
                queue.offer(curr);

            }
        }

        return ans;
    }

    static class Node {
        int prior;
        int idx;

        Node(int prior, int idx) {
            this.prior = prior;
            this.idx = idx;
        }
    }

}