[백준] 1715 - 카드 정렬하기 (힙, 정렬)

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

1715번: 카드 정렬하기

import heapq
 
n = int(input())
q = list()
for _ in range(n):
    heapq.heappush(q, int((input())))
 
if n == 1:
    print(0)
    exit()
 
answer = 0
while len(q) > 1:
    val = heapq.heappop(q) + heapq.heappop(q)
    answer += val
    heapq.heappush(q, val)
 
print(answer)

힙을 이용한 우선순위 큐를 이용하면 되는 정렬 문제이다.

주의할 점은, n이 1 이상이니 n이 1로 들어오면 0을 출력하고 종료되어야 한다.