Lv.3_야근지수

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

프로그래머스

import heapq

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

 

제곱의 합이 최소가 되려면 n이 돌아가며 가장 큰 숫자에서 1을 빼줘야 한다

 

→ 돌아가며 가장 작은 (큰) 숫자 확인 : heapq

 

다만, heappop을 사용할 경우 해당 힙에서 가장 작은 숫자가 반환되므로, -w를 사용해 가장 큰 숫자를 반환하도록 한다

그 이후 i += 1을 사용해 1을 빼주는 역할을 한다

 

heapq를 자주 활용할 수 있도록 꼭 기억해두자