백준 1715번 카드 정렬하기 | python | 우선순위 큐

문제

링크

풀이

계속 더하다보면 heap의 길이가 0이 아니라 1이 된다는 것 잊지 말기

import heapq
n=int(input())

heap=[]
for _ in range(n):
    num=int(input())
    heapq.heappush(heap,num)

result=0
while True:
    if len(heap)==1:
        break
    card1=heapq.heappop(heap)
    card2=heapq.heappop(heap)
    result+=(card1+card2)
    heapq.heappush(heap,card1+card2)
    # print(result,1)
print(result)