[Python] 백준 1715. 카드 정렬하기

 

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

 

 

풀이

힙큐를 이용하여 최소 2개를 pop(heappop)하고 그 값을 누적하면서 힙큐에 다시 넣는다.

 

코드

import heapq, sys
input = sys.stdin.readline
n = int(input())
if n == 1:
    print(0)
    exit()
pq = []
for i in range(n):
    heapq.heappush(pq,int(input()))
res = 0
while len(pq) >= 2:
    temp = 0
    temp += heapq.heappop(pq)
    temp += heapq.heappop(pq)
    res += temp
    heapq.heappush(pq, temp)

print(res)