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

https://programmers.co.kr/learn/courses/30/lessons/42627

코딩테스트 연습 - 디스크 컨트롤러

 

heapq 를 이용하여 문제에서 말하는 대로 구현하면 된다.

현재 시각 T 를 설정하여 정렬된 jobs 의 가장 작은 시점에 있는 값보다 T 가 크다면 heapq 에 넣고 그렇지 않으면, heapq 에서 job 을 꺼내어 T 와 answer 를 재설정한다. 

 

import heapq

def solution(jobs):
    answer = 0
    jobs.sort(key=lambda x : (-x[0], -x[1])) 
    queue = []
    T = 0
    N = len(jobs)
    
    while jobs or queue:
        if ((jobs and T < jobs[-1][0]) or (not jobs)) and queue:
            job = heapq.heappop(queue)
            T += (job[0] if T > job[1] else job[0] + job[1] - T)
            answer += (T - job[1])
        else:
            job = jobs.pop()
            heapq.heappush(queue, job[::-1])
    
    answer = answer // N
    
    return answer

print(solution([[0, 3], [1, 9], [2, 6]])) # 9