[BOJ] 1715번 카드 정렬하기

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

1715번: 카드 정렬하기

import heapq

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


res = 0
while q:
    if len(q) == 1:
        break
    a = heapq.heappop(q)
    b = heapq.heappop(q)
    c = a+b
    res += c
    heapq.heappush(q, c)
    
print(res)

문제의 핵심은 우선순위 큐로 heapq를 사용하는 것이다.

처음에는 그리디인줄 알고 정렬시킨 뒤 앞에서부터 더해 나갔는데, 생각이 짧았었다.