15903 카드 합체 놀이

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

내 풀이(맞음)

import heapq

a, b = map(int,input().split())
ary = input().split()
ary = list(map(int, ary))
heapq.heapify(ary) #리스트를 힙으로 변환
for i in range(b):
    sum = ary[0]
    heapq.heappop(ary)
    sum += ary[0]
    heapq.heappop(ary)
    heapq.heappush(ary, sum)
    heapq.heappush(ary, sum)
sum = 0
for i in ary:
    sum += i
print(sum)

heap을 사용하여 최소값을 2개를 추출하고 삭제한다음 heap에 더하는 방식으로 구현했다.

heapq.heapify(리스트 이름)으로 리스트를 heap으로 변환이 가능하다. 그리고 heapq.heapop와 heapq.heappush(리스트 이름, 값)로 값을 삭제하고 더한다는 것을 다시 유의하면 좋을 것 같다.