[Boj/백준] 15903 Python

문제출처: https://www.acmicpc.net/problem/15903

15903번: 카드 합체 놀이

 

 

1. 문제 접근 방식

힙에 삽입하고 m번 동안 heappop 2회 한 후에 둘을 더한 값을 heappush 2회 해준다.

 

 

2. 내가 푼 코드

import sys
import heapq

n, m = map(int,sys.stdin.readline().split())

heap = []

arr = list(map(int,sys.stdin.readline().split()))

for i in arr:
    heapq.heappush(heap, i)

for i in range(m):
    a = heapq.heappop(heap)
    b = heapq.heappop(heap)
    heapq.heappush(heap,a+b)
    heapq.heappush(heap,a+b)

print(sum(heap))

 

 

3. 결과 및 느낀점

힙 라이브러리 진짜 편리하다...