문제풀이/백준(Boj) 문제풀이

[백준][자료구조/힙] 1715.카드 정렬하기 (파이썬/Python)

얄루몬 2022. 1. 11. 09:32

import heapq
import sys
input = sys.stdin.readline

n= int(input())
h = []
ans = 0
for _ in range(n):
    x = int(input())
    heapq.heappush(h,x)

while len(h) > 1:
    a = heapq.heappop(h)
    b = heapq.heappop(h)
    ans += a+b
    heapq.heappush(h,a+b)
print(ans)

# 계속 answer 을 a+b 대신 넣어서 누적값 때문에 틀렸다고 나왔던 문제 ㅋㅋ .. 정신 차리자