📌 [BOJ] 백준 1715 카드 정렬하기

📖 문제

📖 예제

📖 풀이

import heapq
import sys

N = int(input())
heap = []
for _ in range(N):
  heapq.heappush(heap, int(sys.stdin.readline()))
  
total = 0
while len(heap) > 1:
  f = heapq.heappop(heap)
  s = heapq.heappop(heap)
  sum = f + s
  total += sum
  heapq.heappush(heap, sum)
  
print(total)

heap 자료구조를 이용하면 되는 문제이다.
두 개의 수를 뽑아 더해준 후 total에 더해주고 다시 heap에 넣는다.
이를 heap 안에 수가 1개가 될 때 까지 반복해주고 마지막에 total을 출력한다.