[프로그래머스] 야근 지수(Python)

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

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

heapq를 활용해서 매번 정렬하지 않고 풀면 통과한다.

 

나의 풀이

import heapq
def solution(n, works):
    q = []
    for work in works:
        heapq.heappush(q,-work)
    for _ in range(n):
        if q:
            t = heapq.heappop(q)
            if t !=-1:
                heapq.heappush(q,t+1)
    return sum(list(map(lambda x: x**2,q)))