[Java] 프로그래머스 : 프린터

문제

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

프로그래머스

 

풀이 과정

우선 순위가 1~9까지 있고, 문서가 1~100까지이다. 따라서 문서를 우선순위에 따라 정렬하지 않고, int 1차원 배열에 각 우선 순위가 몇 개 있는지 표시하였다.

그리고나서 우선순위를 9부터 1까지 차례로 탐색하면서, 해당 우선순위를 가진 문서가 1개 이상이면 queue에서 찾도록 했다.

import java.util.*;

class Solution {
    public int solution(int[] priorities, int location) {
        int answer = 0;
        int[] cnt = new int[10];
        Queue<int[]> q = new LinkedList<>();
        
        for(int i=0, size = priorities.length; i<size; i++){
            cnt[priorities[i]]++;
            q.add(new int[]{i, priorities[i]});
        }
        
        int priority = 9;
        while(priority >= 0){
            if(cnt[priority] > 0){
                int[] now = q.poll();
                if(now[1] == priority){
                    cnt[priority]--;
                    answer++;
                    if(now[0] == location){
                        break;
                    }
                }
                else{
                    q.add(now);
                }
            }
            else{
                priority--;
            }
        }
        
        return answer;
    }
}