[알고리즘] 프로그래머스 - 야근 지수
내 풀이
import heapq
def solution(n, works):
works = list(map(lambda x: -x, works))
heapq.heapify(works)
while n:
if works[0] < 0:
work = heapq.heappop(works)
work += 1
heapq.heappush(works, work)
n -= 1
answer = 0
print(works)
while works:
answer += heapq.heappop(works) ** 2
return answer
파이썬에서 내장함수 heap은 min heap이므로 음수를 곱해서 max heap으로 사용했다. 남은 works에서 가장 큰 값들을 줄이면 된다.
Author And Source
이 문제에 관하여([알고리즘] 프로그래머스 - 야근 지수), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@injoon2019/알고리즘-프로그래머스-야근-지수저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)