프로그래머스_프린터(JAVA)

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

프로그래머스

[문제 풀이]

우선순위큐를 이용하여 문제를 풀었습니다.

 

우선순위 작업 배열과 위치가 주어 졌을 때, 해당 작업은 몇번째에 작업이 실행 되는지 찾는 문제입니다.

해당 문제를 풀기 위해서 우선순위큐를 이용해 작업 중요도를 넣어 주었고 배열을 순회하면서 큐와 비교하는 식으로 문제를 해결했습니다.

 

구현한 방식은 아래와 같습니다.

1. 주어진 배열을 역순으로 우선순위큐에 삽이하여 준다.
2. 큐의 앞쪽 원소와 배열의 값을 비교하면서 순회한다.
3. 배열의 값이 큐의 값과 같다면 위치를 비교해주고 위치가 같으면 값을 반환
4. 아니면 큐의 값을 poll해주고 다음 값과 비교한다.
5. 이를 반복한다.

[JAVA]

import java.util.*;

class Solution {
    public int solution(int[] priorities, int location) {

        int answer = 1;
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        
        for(int item: priorities)
            pq.add(item);
        
        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;
    }
}