[프로그래머스] 디스크 컨트롤러 (Heap)-JAVA

[문제 설명]

하드디스크는 한 번에 하나의 작업만 수행할 수 있습니다. 디스크 컨트롤러를 구현하는 방법은 여러 가지가 있습니다. 가장 일반적인 방법은 요청이 들어온 순서대로 처리하는 것입니다.

 

예를들어

- 0ms 시점에 3ms가 소요되는 A작업 요청
- 1ms 시점에 9ms가 소요되는 B작업 요청
- 2ms 시점에 6ms가 소요되는 C작업 요청

와 같은 요청이 들어왔습니다. 들어온 순서대로 값을 정한다고 하면 결과는 다음과 같습니다.

- A: 3ms 시점에 작업 완료 (요청에서 종료까지 : 3ms)
- B: 1ms부터 대기하다가, 3ms 시점에 작업을 시작해서 12ms 시점에 작업 완료(요청에서 종료까지 : 11ms)
- C: 2ms부터 대기하다가, 12ms 시점에 작업을 시작해서 18ms 시점에 작업 완료(요청에서 종료까지 : 16ms)

이 때 각 작업의 요청부터 종료까지 걸린 시간의 평균은 10ms(= (3 + 11 + 16) / 3)가 됩니다.

하지만 A → C → B 순서대로 처리하면

 

- A: 3ms 시점에 작업 완료(요청에서 종료까지 : 3ms)
- C: 2ms부터 대기하다가, 3ms 시점에 작업을 시작해서 9ms 시점에 작업 완료(요청에서 종료까지 : 7ms)
- B: 1ms부터 대기하다가, 9ms 시점에 작업을 시작해서 18ms 시점에 작업 완료(요청에서 종료까지 : 17ms)

이렇게 A → C → B의 순서로 처리하면 각 작업의 요청부터 종료까지 걸린 시간의 평균은 9ms(= (3 + 7 + 17) / 3)가 됩니다.

 

각 작업에 대해 [작업이 요청되는 시점, 작업의 소요시간]을 담은 2차원 배열 jobs가 매개변수로 주어질 때, 작업의 요청부터 종료까지 걸린 시간의 평균을 가장 줄이는 방법으로 처리하면 평균이 얼마가 되는지 return 하도록 solution 함수를 작성해주세요. (단, 소수점 이하의 수는 버립니다)

 

[제한 사항]

• jobs의 길이는 1 이상 500 이하입니다.
• jobs의 각 행은 하나의 작업에 대한 [작업이 요청되는 시점, 작업의 소요시간] 입니다.
• 각 작업에 대해 작업이 요청되는 시간은 0 이상 1,000 이하입니다.
• 각 작업에 대해 작업의 소요시간은 1 이상 1,000 이하입니다.
• 하드디스크가 작업을 수행하고 있지 않을 때에는 먼저 요청이 들어온 작업부터 처리합니다.

[풀이]

문제 해결을 위해서 알아야할 핵심적인 부분은 다음과 같다. 

• 주어진 jobs는 정렬된 상태가 아니기 때문에 요청시점에 따라 오름차순으로 정렬
• 먼저 실행된 디스크가 시작되고 끝나는 동안 요청된 디스크들을 소요시간 별로 정렬하기 위해 Priority 사용
• Priority 안에 있는 값들을 순서대로 꺼내면서 다음 디스크의 범위와 전체 결과를 위한 answer 값을 더해 나감

import java.util.*;
class Solution {
    public int solution(int[][] jobs) {
        Arrays.sort(jobs,((o1,o2)->o1[0]-o2[0]));
        PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);
        
        int k=0;
        int t = jobs[0][0];
        int answer=0;
        int count=0;
        
        while (count < jobs.length) {
            while(k<jobs.length&&jobs[k][0]<=t){
                pq.add(jobs[k++]);
            }

            if(pq.size()==0){
                t = jobs[k][0];
            }else{
                int temp[] = pq.poll();
                answer += temp[1] + t - temp[0];
                t += temp[1];
                count++;
            }

        }
        return answer/jobs.length;

    }
}