🟩 [백준] [Python] [Gold4] 1715번_카드 정렬하기

문제

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

1715번: 카드 정렬하기

풀이

# 백준 1715번 문제 - 카드 정렬하기
import heapq
import sys
input = sys.stdin.readline

n = int(input())
cards = []
result = 0

for i in range(n):
    heapq.heappush(cards, int(input()))

if len(cards) == 1:
    print(0)
else:
    while len(cards) > 1:
        plus = heapq.heappop(cards) + heapq.heappop(cards)
        result += plus
        heapq.heappush(cards, plus)
    
    print(result)

힙 자료구조를 사용하여 풀면 간단한 문제이다.

힙 문제를 많이 풀어보지 않아서 처음에 생각하는데 시간이 오래 걸렸다.

힙 자료구조에 대해 정리하는 글을 쓰도록 해야겠다.