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

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

프로그래머스

문제 정보

문제 유형	난이도	걸린 시간	해결 유무	스스로 구현 성공
힙	level3	45m	o	x

코드

처음 접근했던 방식

• 길이가 가장 작은 것들(jobs[][1])로 정렬한 후 계산해주면 된다고 생각

def solution(jobs):
    answer, total = 0, 0
    jobs.sort(key=lambda x : x[1])
    total = jobs[0][1] - jobs[0][0]
    answer = total 
    for i in range(1, len(jobs)):
        answer += total + jobs[i][1] - jobs[i][0]
        total += jobs[i][1]
    return answer//len(jobs)

반례 : [[0,3],[4,1]]

jobs[][0]도 고려해줘야 하는 요소임을 생각 못함

 

정답 코드

from heapq import *

def solution(jobs):
    answer, i, total = 0,0,0
    l = len(jobs)
    h = []

    while i < l:
        for j in jobs:
            if 0 <= j[0] <= total:
                heappush(h, (j[1], j[0]))
                j[0] = -1
        if h:
            cur = heappop(h)
            total += cur[0]
            answer += total - cur[1]
            i += 1
        else:
            total += 1
    return answer//l

heapq를 사용해서 문제 해결