04_01 농심 라면 공장

1) 버틸 수 있는 최대한에서 공급날짜 대상 정하기
2) 공급 날짜 대상 중에서 최댓값을 맥스 힙으로 뽑기
3) stock이 k 보다 작은 경우 반복하기

import heapq

ramen_stock = 4
supply_dates = [4, 10, 15]
supply_supplies = [20, 5, 10]
supply_recover_k = 30

def get_minimum_count_of_overseas_supply(stock, dates, supplies, k):
answer = 0
last_added_date_index = 0
max_heap = []

while stock <= k:
    while last_added_date_index < len(dates) and dates[last_added_date_index] <= stock:
        heapq.heappush(max_heap, -supplies[last_added_date_index])
        last_added_date_index += 1

    answer += 1
    heappop = heapq.heappop(max_heap)
    stock += -heappop

return answer

print(get_minimum_count_of_overseas_supply(ramen_stock, supply_dates, supply_supplies, supply_recover_k))
print("정답 = 2 / 현재 풀이 값 = ", get_minimum_count_of_overseas_supply(4, [4, 10, 15], [20, 5, 10], 30))
print("정답 = 4 / 현재 풀이 값 = ", get_minimum_count_of_overseas_supply(4, [4, 10, 15, 20], [20, 5, 10, 5], 40))
print("정답 = 1 / 현재 풀이 값 = ", get_minimum_count_of_overseas_supply(2, [1, 10], [10, 100], 11))

좋은 웹페이지 즐겨찾기