[백준][Python] 15903번 카드 합체 놀이 - 실버 1

https://www.acmicpc.net/problem/15903

15903번: 카드 합체 놀이

 

문제

 

풀이

주어진 카드들 중에서 가장 작은 숫자의 카드를 2장 뽑아서 합한 수를 두 카드의 숫자에 새롭게 넣어주면 된다.

min()으로 구현해줘도 되지만, heapq르 활용하여 우선순위 큐 방법으로 풀어주었습니다.

 

코드

# 15903번 카드 합체 놀이
from heapq import heappush, heappop, heapify
n , m = map(int, input().split())
cards = list(map(int, input().split()))
heapify(cards)

for _ in range(m):
    card1 = heappop(cards)
    card2 = heappop(cards)
    heappush(cards, card1+card2)
    heappush(cards, card1+card2)

print(sum(cards))

 

출력결과