[백준] 15903 파이썬(python) : 카드 합체 놀이

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

15903번: 카드 합체 놀이

 

 

import sys, heapq
n, m = map(int, sys.stdin.readline().split())
heap = list(map(int, sys.stdin.readline().split())) #1
heapq.heapify(heap) #2

for _ in range(m):
    n1 = heapq.heappop(heap) #3
    n2 = heapq.heappop(heap)
    tmp = n1 + n2 #4
    heapq.heappush(heap, tmp) #5
    heapq.heappush(heap, tmp)
print(sum(heap)) #36

 

#1 : 입력받을 숫자를 리스트로 받는다

#2 : 그리고 일반 리스트를 힙 자료구조로 바꾼다

#3 : 최소 힙을 보장하므로 원소 2개를 꺼낸다

#4 : 꺼낸 원소를 더한 값을 임시로 생성하고

#5 : 두번 넣어준다

#6 : 과정이 끝나고 힙 자료구조에 있는 원소를 모두 더한다