[프로그래머스] / [Level 3] / [Python] 야근 지수

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

프로그래머스

---

heapq 모듈을 사용하여 해결하였다.

import heapq

def solution(n, works):
    answer = 0
    
    if sum(works) <= n:
        return 0
    
    works = [-i for i in works]
    heapq.heapify(works)
    # 야근을 한 시점에서 남은 일의 작업량을 제곱하여 더한 값
    while n != 0:
        p = heapq.heappop(works)
        p += 1
        n -= 1        
        heapq.heappush(works, p)
    
    for work in works:
        answer += work * work
    
    return answer