카테고리 없음

백준 파이썬 카드 정렬하기

작지 2022. 4. 27. 01:10

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

 

1715번: 카드 정렬하기

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

www.acmicpc.net

 

최솟값들을 계속해서 합치고 넣어주고를 반복해주면 되는 문제였습니다.

 

우선순위 큐로 야무지게 뺴고 합치고 집어넣으면 끝

 

import heapq
import sys
input = sys.stdin.readline

n = int(input())

heap = []

for i in range(n):
    x = int(input())
    heapq.heappush(heap,x)

if n == 1:
    print(0)
else:
    result = 0
    while len(heap) > 1:
        x_1 = heapq.heappop(heap)
        x_2 = heapq.heappop(heap)

        result += x_1 + x_2

        heapq.heappush(heap,(x_1+x_2))
    print(result)