디스크 컨트롤러

프로그래머스

• 하드디스크는 한 번에 하나의 작업만 수행할 수 있습니다. 디스크 컨트롤러를 구현하는 방법은 여러 가지가 있습니다. 가장 일반적인 방법은 요청이 들어온 순서대로 처리하는 것입니다.
• 각 작업에 대해 [작업이 요청되는 시점, 작업의 소요시간]을 담은 2차원 배열 jobs가 매개변수로 주어질 때, 작업의 요청부터 종료까지 걸린 시간의 평균을 가장 줄이는 방법으로 처리하면 평균이 얼마가 되는지 return 하도록 solution 함수를 작성해주세요. (단, 소수점 이하의 수는 버립니다)
• jobs의 길이는 1 이상 500 이하입니다.
• jobs의 각 행은 하나의 작업에 대한 [작업이 요청되는 시점, 작업의 소요시간] 입니다.
• 각 작업에 대해 작업이 요청되는 시간은 0 이상 1,000 이하입니다.
• 각 작업에 대해 작업의 소요시간은 1 이상 1,000 이하입니다.
• 하드디스크가 작업을 수행하고 있지 않을 때에는 먼저 요청이 들어온 작업부터 처리합니다.

입출력

jobs	return
[[0, 3], [1, 9], [2, 6]]	9

---

접근 방식

현재 시점에서 처리할 수 있는 작업들을 모아 놓고 하나를 뽑아 현재 시점과 총 대기시간을 구해주는 것을 반복

• 작업의 소요 시간 기준으로 최소힙을 만든다.
• 현재 시점에서 처리할 수 있는지의 여부는, 작업의 요청 시간이 이전에 완료한 작업의 시작 시간과 현재 시점 사이에 있어야한다.
• 만약 현재 처리할 수 있는 작업이 없다면 남아있는 작업들의 요청 시간이 아직 오지 않았으므로 현재 시점을 업해준다.

코드

import heapq

def solution(jobs):
    q = []
    ans, i = 0, 0
    prev_start = -1 # 이전 시작 시각
    current = 0 # 현재 시각

    while i < len(jobs):
        for start, period in jobs:
            # 이전 작업의 시작과 현재 사이에 시작할 수 있는 작업이 있으면 힙에 넣어준다.
            if prev_start < start <= current:
                # 작업이 걸리는 기간이 짧은 순대로 진행해야하므로
                heapq.heappush(q, (period, start))
        
        # 현재 시작할 수 있는 작업이 있는 경우
        if len(q) > 0:
            # 작업 시작
            now_p, now_s = heapq.heappop(q)
            prev_start = current # 현재가 이전이 되고
            current += now_p # 지금 시각에 걸리는 시간 더하고 = 현재 
            ans += (current-now_s) # 현재에서 시작했던 시간을 빼면 작업하는데에 걸린 시간
            i += 1
        else:
            # 작업할 게 당장 없으면 시간만 간다
            current += 1
            
    return int(ans/len(jobs))