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

문제

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

 

문제 풀이

 하드 디스크의 요청된 작업을 작업이 종료되는 시간이 빠른 순으로 처리하도록 하여 평균 처리 시간을 반환하는 문제이다. 요청된 작업 순으로 정렬 후에 처리할 작업을 하나씩 `pop`하여 `heap`을 통해 처리 시간이 가장 큰 작업부터 루트가 되도록 유지한다. 이와 같은 과정을 통해, 문제에 요구하는 것과 같이 작업을 처리할 수 있다.

 

코드

import heapq
from collections import deque

RQEUST = 0

def solution(jobs):
    answer = 0
    jobs.sort()
    q = deque(jobs)
    current_time = q[0][RQEUST]
    num_jobs = len(q)
    heap = []

    while q:
        start, process = q.popleft()

        current_time += process
        answer += current_time - start

        while q and q[0][RQEUST] < current_time:
            start, process = q.popleft()
            heapq.heappush(heap, [-process, start])

        while heap:
            process, start = heapq.heappop(heap)
            q.appendleft([start, -process])
    return answer // num_jobs