Heap 자료구조
이진트리 구조로 구성되어 있으며, 추가/삭제의 시간 복잡도는 O(logN)이다.
문제
더 맵게
https://programmers.co.kr/learn/courses/30/lessons/42626
문제
매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다.
섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)
Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다.
Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요.
제한사항
scoville의 길이는 2 이상 1,000,000 이하입니다.
K는 0 이상 1,000,000,000 이하입니다.
scoville의 원소는 각각 0 이상 1,000,000 이하입니다.
모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다
예시
## scoville K return
[1, 2, 3, 9, 10, 12] 7 2
풀이
새로운 스코빌 지수를 계산하여 추가하고 sorting을 하면 N*NlogN 으로 시간 복잡도에 걸린다. Heap을 이용하면 배열에 추가 및 sorting 까지 O(logN) 을 가지므로 최악의 경우 (예를들어, K가 1,000,000,000 이고 scoville 배열에 1만 1,000,000개일 경우) O(NlogN) 이다.
import heapq
def solution(scoville, K):
# 모든 음식 스코빌 지수 K 이상 => 섞어야됨
# sorting으로 풀면 복잡도 N * NlogN
# heap을 활용하면 맨 앞 2개는 무조건 정렬되어 있을 것이므로 heap 활용
answer = 0
heapq.heapify(scoville)
while scoville[0] <= K and len(scoville) >= 2:
first_value = heapq.heappop(scoville)
second_value = heapq.heappop(scoville)
heapq.heappush(scoville, first_value+(second_value*2))
answer += 1
# 다 합쳤는데 K보다 작은 경우는 -1 return
if len(scoville) < 2 and scoville[0] < K:
return -1
return answer