[프로그래머스] 프린터

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

코딩테스트 연습 - 프린터

 

PriorityQueue 를 이용하여 중요도가 높은 순으로 인쇄(poll) 해준다. 대기목록에 우선순위가 가장 큰 값을 아래와 같이 for 를 통하여 차례대로 비교하면 중요도가 같은 문서에 대해서도 조건에 알맞게 답을 구할 수 있다.

 

import java.util.*;

class Solution {
    public int solution(int[] priorities, int location) {
        int answer = 1;
        PriorityQueue<Integer> queue = new PriorityQueue<>(Comparator.reverseOrder());
        
        for (int i=0;i<priorities.length;i++) {
            queue.add(priorities[i]);
        }
        
        while (queue.size() != 0) {
            for (int i=0;i<priorities.length;i++) {
                if (priorities[i] == queue.peek()) {
                    if (i == location) {
                        return answer;
                    } else {
                        queue.poll();
                        answer++;
                    }
                }
            }
        }
        
        return answer;
    }
}