[Python] 프로그래머스 <야근 지수>

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

프로그래머스

 

- 효율성 체크 실패 (시간 초과)

def solution(n, works):
    answer = 0
    
    works.sort()
    for _ in range(n):
        works.append(works.pop()-1)
        works.sort()
    
    for i in works:
        if i > 0:
            answer += i*i
    return answer

 

- 정답 코드

import heapq

def solution(n, works):
    answer = 0
    # heapq가 최소힙이기 때문에
    # 작업량이 가장 많은 작업을 pop하기 위해 최대힙으로 바꿈
    works = [-work for work in works]
    heapq.heapify(works)
    
    # 작업량이 많은 작업을 1씩 빼기
    for _ in range(n):
        num = heapq.heappop(works)
        heapq.heappush(works, num+1)
        
    for i in works:
        if i < 0:
            answer += i*i    
    return answer