[P&B] #25 Programmers

프로그래머스 Lv.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 … 과 같이 표현합니다.

 

나의 문제 풀이 코드 (45.0/100)

import java.util.*;

class Solution {
    public static int solution(int[] priorities, int location) {
        int answer = 0;

        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if (o1[0] == o2[0])
                    return Integer.compare(o2[1], o1[1]);
                return Integer.compare(o2[1], o1[1]);
            }
        });
        for (int i = 0; i < priorities.length; i++) {
            priorityQueue.offer(new int[]{i,priorities[i]});

        }

        int count = 1;
        while (!priorityQueue.isEmpty()){
            int[] current = priorityQueue.poll();
            if (current[0] == location){
                answer = count;
                break;
            }
            count++;
        }


        return answer;
    }
}

 

답안 참고 후 수정한 코드

import java.util.*;

class Solution {
    public int solution(int[] priorities, int location) {
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder()); //우선순위 큐를 내림차순 정렬로 설정한다.
        int answer = 0;

        for (int i = 0; i < priorities.length; i++) {
            pq.add(priorities[i]);
        }

        while (!pq.isEmpty()) {
            for (int i = 0; i < priorities.length; i++) {
                if (priorities[i] == pq.peek()) { //priorities의 값이 현재 오름차순 우선순위가 제일 높은 값과 같은지 확인한다.
                    if (i == location) {
                        answer++;
                        return answer;
                    }   
                    pq.poll();
                    answer++;
                }
            }
        }
        return -1;
    }
}

 

문제 풀이 코멘트

정말 놀랍게도(?) 문제를 많이 안풀어본건지 풀어봤는데 기억을 못하는 것인지 우선순위 큐 개념은 알았지만 문제로 풀어본것은 이번이 처음이다. 그래서 우선순위 큐 구현자체가 어려웠고 어떤식으로 풀어가야할지 감을 잡지 못해 빙빙 둘러가듯 문제를 풀었다. 테스트코드 상으로는 맞았지만 제출을 했을 때 반도 못맞추어 당황스러웠다. (나름 완벽하다고 생각했나보다) 답안을 보고나니 꽤 머리를 많이 쓰려고 노력한 티는 난 코드였단게 느껴졌다.