프로그래머스: 야근 지수

문제

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

 

문제 풀이

 남은 일의 작업량이 있을 때, 야근 지수를 최소화하여 반환하는 문제이다. 일의 작업량이 있고, 일을 할 수 있는 시간이 있다면 작업량이 가장 큰 작업부터 처리하여야 한다. 야근 지수는 작업량의 제곱이기 때문이다.

 

 문제를 풀기 위해서는 Python의 경우 heapq를 사용하여 풀어야 한다. heapq는 최소 힙이기 때문에 최대 힙을 만들기 위해서는 음수 값으로 변환하여야 한다. (이는 works가 1 이상 이기에 가능하다.)

 

코드

from heapq import heapify, heappush, heappop


def solution(n, works):
    answer = 0
    works = [-work for work in works]
    heapify(works)
    
    while n:
        max_time = -heappop(works)
        
        if not max_time:
            break
            
        heappush(works, -(max_time - 1))
        n -= 1

    return sum(work ** 2 for work in works)