[우선순위큐] 프린터 (자바)

https://school.programmers.co.kr/learn/courses/30/lessons/42587/solution_groups?language=java&type=my 

프로그래머스

 

  문서의 우선순위(중요도)가 담긴 priorities 배열과, 특정 문서의 위치 location 이 주어진다. 이때 location 에 위치한 문서는 실제로 몇 번째로 출력이 되는지 answer 에 반환해주면 된다. 이를 위해 우선순위큐를 사용한다. 숫자가 높을 수록 중요도가 높다고 문제에서 제시하였으므로, PriorityQueue 에 Collections.reverseOrder() 을 전달하여 중요도가 높은 순서대로 poll 이 될 수 있도록 한다. 

 

  그리고 priorities 배열을 순회하며, 현재 우선순위 큐에 저장된 가장 높은 중요도 == priorities[i] 라면 해당 위치(i) 가 location 인지 확인하고 맞다면 answer 을 반환한다. 해당 위치(i) 가 location 이 아니라면 answer 을 하나 증가시켜주고 pop/poll 해준다. location 에 위치한 문서에만 관심이 있기 때문이다.

 

import java.io.IOException;
import java.util.Collections;
import java.util.PriorityQueue;

class Solution {
    public int solution(int[] priorities, int location) throws IOException {
        int answer = 1;
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        for(int ele : priorities)
            pq.add(ele);

        while(!pq.isEmpty()){
            for(int i = 0 ; i < priorities.length; i++){
                if(pq.peek() == priorities[i]){
                    if(i == location) return answer;
                    else {
                        answer++;
                        pq.poll();
                    }
                }
            }
        }
        return answer;
    }
}