[백준] 1715 카드 정렬하기 (파이썬 python)

 

 

문제 링크

 

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

1715번: 카드 정렬하기

 

 

문제 풀이

 

import heapq

nums = []
result = 0
n = int(input())
for _ in range(n):
    heapq.heappush(nums, int(input()))
while len(nums) > 1:
    n1 = heapq.heappop(nums)
    n2 = heapq.heappop(nums)
    heapq.heappush(nums, n1 + n2)
    result += (n1 + n2)
print(result)

 

 

매번 가장 작은 두 숫자를 더하면 최솟값을 도출할 수 있다.

항상 정렬된 상태를 유지해야하므로 heapq를 사용했다.