[백준] 카드 정렬하기

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

1715번: 카드 정렬하기

 

우선순위 큐(heapq) 를 이용해서 가장 작은 2개의 숫자의 합을 계산해주고 그 값을 또 queue 에 넣어준다. 이러한 과정을 반복하여 queue 에는 하나의 값만 남을 때까지 계속한다. 마지막으로 숫자의 합을 계속 더해준 answer 를 return 한다. 

 

import sys
import heapq

answer = 0
N = int(sys.stdin.readline())
queue = []
for i in range(N):
    heapq.heappush(queue, int(sys.stdin.readline()))

while len(queue) > 1:
    AB = 0

    for i in range(2):
        AB += heapq.heappop(queue)

    answer += AB
    heapq.heappush(queue, AB)

print(answer)