[프로그래머스]Level3 - 디스크컨트롤러

package level3;
import java.util.*;
public class Solution_디스크컨트롤러 {
	public static void main(String[] args) {
		int[][] jobs = {{0, 5}, {2, 2}, {4, 2}};
		System.out.println(solution(jobs));
	}
//  테스트 케이스
//	[[0, 3], [4, 3], [8, 3]] 3
//	[[0, 5], [6, 1], [6, 2]] 3
//	[[0, 5], [6, 2], [6, 1]] 3
//	[[0, 5], [2, 2], [5, 3]] 5
//	[[0, 5], [2, 2], [4, 2]] 5

	static public int solution(int[][] jobs) {
		// 작업시작시간 오름차순으로 정렬한다.
		Arrays.sort(jobs, (o1, o2) -> o1[0]-o2[0]);
		Queue<Job> pq = new PriorityQueue<>();
		int answer = 0;
		int work = 0;
		int index = 0;
		
		while(work < jobs.length) {
			// 현재 시간 이하의 작업들을 모두 PriorityQueue에 넣는다.
			// 작업시간이 가장 작은 작업이 맨 앞에 위치한다.
			while(index < jobs.length && jobs[index][0] <= time) {	
				pq.add(new Job(jobs[index][0], jobs[index++][1]));
			}
			
			// 현재 시간 이하의 작업이 없다면, 첫 번째 작업 시작시간을 현재 시간으로 시작.
			if(pq.isEmpty()) {
				time = jobs[index][0];
			}else {
				// 현재 시간보다 작은 작업중 가장 작은 크기의 작업을 가져온다.
				Job j = pq.poll();
				answer += time - j.start + j.job;
				time += j.job;
				work++;
			}
		}
		
        return answer/jobs.length;
    }
	static int time;
	
	static class Job implements Comparable<Job>{
		int start, job;

		public Job(int start, int job) {
			this.start = start;
			this.job = job;
		}

		@Override
		public int compareTo(Job o) {
			return this.job - o.job;
		}
		
		@Override
		public String toString() {
			return start + "," + job;
		}
	}
}