◦ Algorithm/Python

백준 카드정렬하기 1715 그리디

밍블리s2 2023. 4. 22. 17:47

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

 

1715번: 카드 정렬하기

정렬된 두 묶음의 숫자 카드가 있다고 하자. 각 묶음의 카드의 수를 A, B라 하면 보통 두 묶음을 합쳐서 하나로 만드는 데에는 A+B 번의 비교를 해야 한다. 이를테면, 20장의 숫자 카드 묶음과 30장

www.acmicpc.net

 

 

  • sys로 입력받아야 속도 빨라짐
  • sum과 answer 각각 만들어줘야 함
import heapq
from sys import stdin

cards = []
answer = 0
for _ in range(int(stdin.readline())):
    heapq.heappush(cards, int(stdin.readline()))

while len(cards)>1:
    sum = heapq.heappop(cards) + heapq.heappop(cards)
    answer += sum
    heapq.heappush(cards, sum)

print(answer)