백준 알고리즘 15903. 카드 합체 놀이

스마트코더91 2024. 3. 12. 20:39

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

15903번: 카드 합체 놀이

첫 번째 줄에 카드의 개수를 나타내는 수 n(2 ≤ n ≤ 1,000)과 카드 합체를 몇 번 하는지를 나타내는 수 m(0 ≤ m ≤ 15×n)이 주어진다. 두 번째 줄에 맨 처음 카드의 상태를 나타내는 n개의 자연수 a1,

www.acmicpc.net

전체 코드

import heapq

n, m = map(int, input().split())
heap = list(map(int, input().split()))
# for n in nums:
#     heapq.heappush(heap, n)
heapq.heapify(heap)

for _ in range(m):
    n1 = heapq.heappop(heap)
    n2 = heapq.heappop(heap)
    s = n1 + n2
    heapq.heappush(heap, s)
    heapq.heappush(heap, s)

print(sum(heap))