[Programmers] 야근 지수

태환·2024년 3월 22일
0

Coding Test

목록 보기
139/151
post-custom-banner

📌 [Programmers] 야근 지수

📖 문제

📖 예제

📖 풀이

def solution(n, works):
    import heapq
    
    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])

여러번의 반복동안 가장 큰 수에 연산을 수행해야 할 경우 heapq.heapify()를 다음과 같이 활용하여 문제를 해결할 수 있다.

profile
연세대학교 컴퓨터과학과 석사 과정
post-custom-banner

0개의 댓글