[Level3] 야근 지수

🛠 문제

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

👩🏻‍💻 해결 방법

최대힙을 사용해 쉽게 풀 수 있는 문제였다
최대힙은 최소힙과 반대라는 것을 숙지하는게 중요했다

import heapq
def solution(n, works):
    if n >= sum(works):
        return 0
    
    works = [-i for i in works]
    heapq.heapify(works)
    for _ in range(n):
        w = heapq.heappop(works) + 1
        heapq.heappush(works, w)
        
    return sum([i**2 for i in works])