프로그래머스: 야근 지수

문제

코딩테스트 연습 - 야근 지수

 

문제 풀이

 남은 일의 작업량이 있을 때, 야근 지수를 최소화하여 반환하는 문제이다. 일의 작업량이 있고, 일을 할 수 있는 시간이 있다면 작업량이 가장 큰 작업부터 처리하여야 한다. 야근 지수는 `작업량의 제곱`이기 때문이다.

 

 문제를 풀기 위해서는 `Python`의 경우 `heapq`를 사용하여 풀어야 한다. `heapq`는 최소 힙이기 때문에 최대 힙을 만들기 위해서는 음수 값으로 변환하여야 한다. (이는 works가 1 이상 이기에 가능하다.)

 

코드

from heapq import heapify, heappush, heappop


def solution(n, works):
    answer = 0
    works = [-work for work in works]
    heapify(works)
    
    while n:
        max_time = -heappop(works)
        
        if not max_time:
            break
            
        heappush(works, -(max_time - 1))
        n -= 1

    return sum(work ** 2 for work in works)