[JAVA/우선순위 큐] 프로그래머스 프린터

1. 문제

https://programmers.co.kr/learn/courses/30/lessons/42587?language=java

코딩테스트 연습 - 프린터

 

2. 풀이

우선순위 큐를 사용해서 풀었다.

1.  우선순위 큐를 사용하면 숫자가 오름차순으로 자동으로 큐에 정렬되는데, 이 문제에서는 우선순의 큐를 내림차순으로 정렬한다.
2. pq가 빌 때 까지(= 모든 문서가 출력될 때 까지), for문을 돌아가며 pq 내의 가장 큰 값과 그때의 위치를 찾는다.
3. pq 내의 가장 큰 값과 그때의 위치를 찾으면, 출력 가능한 조건이므로 출력처리를 한다. 그때의 위치가 찾고자 하는 위치와 동일하면 정답

 

3. 코드

 

import java.util.*;

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

        // 내림차순 우선순위 큐
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>(Collections.reverseOrder());
        for(int task : priorities){
            pq.add(task);
        }

        while(!pq.isEmpty()){
            for(int i = 0; i < priorities.length; i++){
                if(priorities[i] == pq.peek()) { // 출력되지 않은 값중 가장 크면 출력 처리
                    if(i == location){
                        return answer;
                    }
                    pq.poll();
                    answer++;
                }
            }
        }

        return answer;
    }
}