24-gjsk132

[gjsk132]

🔗 문제 링크

백준 15903 : 카드 합체 놀이

✔️ 소요된 시간

20분

✨ 수도 코드

🎯 문제 이해

알고리즘 유형 : 그리디 (greedy)

문제 요약
주어진 카드에는 자연수가 쓰여져 있다.
자연수 카드 중에서 2개를 더한다.
더한 값을 이전의 값에 덮어준다.
카드가 합쳐지는 횟수가 이후에, 카드의 총 합이 가장 작은 수를 구한다.

예시
주어진 카드 : 2, 3, 6
합체 > 가장 작은 ( 2, 3 ) > 2 + 3 = 5
1회 합체한 카드 : 5, 5, 6
결과 : 5 + 5 + 6 = 16

문제 접근
합체하는 과정을 생각해보면, 선택된 자연수가 한 번씩 더해진다고 생각하면 된다.

선택 숫자 : 2, 3
선택한 수의 합 : 5

결과 : 2+3, 3+2
결과의 합 : 10

이 점을 생각했을 때 합체 후 카드의 총 합이 가장 작으려면,
매 단계마다 가장 작은 두 자연수를 합치면 된다는 것을 알 수 있다.

그리디 알고리즘 접근
단계에서 최적( 가장 작은 두 자연수 )의 값을 매번 선택한다는 점에서 그리디 알고리즘을 사용하였다.

힙 자료구조 접근
매번 갱신되는 숫자의 리스트 일일히 작은 값 2개를 구하는 방법도 있지만 생각만 해도 복잡하다.
가장 작은 자연수 2개를 쓴다는 점에서, 힙을 사용해주면 훨씬 수월하게 풀 수 있다.

🔍코드 설명

card_cnt : 카드의 수
merge_cnt : 카드를 합치는 횟수
card_num : 카드의 수를 기억하는 힙 (heap)

입력되는 값들은 각각 해당하는 변수에 넣어준다.

card_cnt, merge_cnt = map(int, input().split())
card_num = list(map(int,input().split()))

card_num에서 가장 작은 값 2개를 빼줘야하기 때문에, 리스트를 힙으로 만들어준다.

hp.heapify(card_num)

핵심 코드
카드들 중에서 가장 작은 값 2개를 빼고, 그 값의 합을 2번 넣어준다.
해당 과정을 반복문을 사용하여 주어진 합체 횟수(merge_cnt)만큼 진행한다.

for i in range(merge_cnt):
    
    a, b = hp.heappop(card_num), hp.heappop(card_num)
    hp.heappush(card_num, a+b)
    hp.heappush(card_num, a+b)

위 과정을 진행하고 나면, 합이 가장 작은 조합으로 수가 나타나져 있을 것이다.

따라서 결과의 합을 출력해주면 된다.

print(sum(card_num))

전체 코드

import heapq as hp

input = open(0).readline

card_cnt, merge_cnt = map(int, input().split())

card_num = list(map(int,input().split()))

hp.heapify(card_num)

for i in range(merge_cnt):
    
    a, b = hp.heappop(card_num), hp.heappop(card_num)
    hp.heappush(card_num, a+b)
    hp.heappush(card_num, a+b)

print(sum(card_num))

📚 새롭게 알게된 내용

힙을 오랜만에 쓰는 것 같아서 복습용으로 좋았습니다!

[9kyo-hwang]

오잉 실버1 치곤 문제가 많이 쉽네요
우선순위 큐의 개념만 알면 뚝딱 해치울 수 있네요? 심지어 큐에 값이 하나만 남는 그런 경우도 없어서 별도의 예외처리를 해줄 필요도 없고...

from heapq import *
input = open(0).readline

_, m = map(int, input().split())
heapify(a := list(map(int, input().split())))

for _ in range(m):
    x, y = heappop(a), heappop(a)
    heappush(a, x + y)
    heappush(a, x + y)
    
print(sum(a))

---

야근 지수 이 문제가 결이 좀 비슷한데 부족하면 요것도 풀어봐요 :)

[Dolchae]

우옹 힙으로 구현하면 이렇게 간단하게 풀 수 있군요..! 저는 하나하나 구해서 시간초과나고 난리났을 것 같은데 아주 좋은 것 같아요😄 수고하셨습니다!!😊

[xxubin04]

힙에 대해 까먹어가던 중에 다시 복습할 수 있었습니다!!
사실 문제를 보고 힙을 써야겠다는 아이디어가 떠오르지 않아서 조금 헤맸습니다..😂
간단명료하게 짜신 코드 보면서 항상 많이 배워갑니다!!😊👍