[백준] 파이썬 1715 : 카드 정렬하기

 

 

 

from heapq import *
n = int(input())
heap = [int(input()) for _ in range(n)] # 요소들 힙에 추가 
heapify(heap)
result = 0 
while len(heap) > 1: # 힙에 더이상 더할 요소가 남지 않을때까지
    x = heappop(heap) # 첫번째 작은 값
    y = heappop(heap) # 두번째 작은 값
    heappush(heap, x + y) # x + y 의 합을 다시 힙에 추가
    result += x + y
print(result)

 

 

문제 그대로 코드를 썼다. 최소값 순으로 두 값을 더하고 그 합을 남은 값과 다시 더하는 방식으로 구현했다. 힙에 더이상 더할 값이 없을때까지 반복문을 돌리게 하고 첫번째로 작은 값과 두번째로 작은 값을 더해주었다. 그리고 그 합을 다시 힙에 추가하여 반복하게 했다. 

 

 

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

1715번: 카드 정렬하기