[프로그래머스] 알고리즘 연습 문제 : 프린터 (자바/JAVA)

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

코딩테스트 연습 - 프린터 | 프로그래머스

 

import java.util.Collections;
import java.util.PriorityQueue;
 
class Solution {
    public int solution(int[] priorities, int location) {
       int answer = 0;
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
 
        for(int priority : priorities) {
            pq.offer(priority);
        }
 
        while(!pq.isEmpty()) {
 
            for(int i = 0; i < priorities.length; i++) {
                if(pq.peek() == priorities[i]) {
                    pq.poll();
                    answer++;
                    if (location == i) {
                        pq.clear();
                        break;
                    }
                }
            }
        }
 
        return answer;
    }
}

 

자바에서 제공하는 우선순위 큐(Priority Queue) 유틸은 힙 기반으로 만들어졌다.

그래서 최대힙, 최소힙을 활용 가능해서 알고리즘 문제에서 종종 눈에 띄곤한다.

처음엔 큐(Queue)를 사용해서 처음부터 확인하고 다시 맨 뒤로 넣고 확인하고 아니면 뒤로...

이런식으로 정말 문제 그대로 만들려다보니 점점 미궁에 미궁에 미궁으로 빠졌다.

간단하게 우선순위 큐에서 높은 우선 순위가 있으면 출력하고 갯수 체크만 하면 끝인걸...

ps. Priority Queue는 그냥 만들면 최소값 기준으로 우선순위 된다.

new Priority Queue(Collections.reverseOrder())로 생성해주게 되면 최대값 기준으로 우선순위가 정렬된다.