백준 & 프로그래머스

프로그래머스 [PCCP 모의고사 #2] 2번 - 신입사원 교육

concho 2023. 12. 3. 12:58
# 임무: 2명 선발해서 공부하기 + 능력치 최소화
# 항상 최소만 뽑는다면 => 능력치 최소화
import heapq
def solution(ability, number):
    heapq.heapify(ability)
    for _ in range(number):
        new_ability = heapq.heappop(ability) + heapq.heappop(ability)
        heapq.heappush(ability, new_ability)
        heapq.heappush(ability, new_ability)
    return sum(ability)