[프로그래머스] 프로세스 - Java

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

프로그래머스

 

프로그래머스 프로세스 자바 풀이

난이도: Lv2

 

풀이

우선순위가 높은 순서로 우선순위 큐에 저장한다.

그 후 우선 순위가 일치할 시 큐에서 꺼낸다.

 

반복하면서 몇 번째 순위에 꺼냈는지 리턴한다.

 

코드

import java.util.Collections;
import java.util.PriorityQueue;

class Solution {
    public int solution(int[] priorities, int location) {
        int answer = 0;
        PriorityQueue<Integer> pQueue = new PriorityQueue<>(Collections.reverseOrder());

        for(int n:priorities){
            pQueue.add(n);
        }

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