LeetCode-IPO|最低加油次數-堆的應用

1、題目:IPO

假設 力扣(LeetCode)即將開始其 IPO。爲了以更高的價格將股票賣給風險投資公司,力扣 希望在 IPO 之前開展一些項目以增加其資本。 由於資源有限,它只能在 IPO 之前完成最多 k 個不同的項目。幫助 力扣 設計完成最多 k 個不同項目後得到最大總資本的方式。

給定若干個項目。對於每個項目 i,它都有一個純利潤 Pi,並且需要最小的資本 Ci 來啓動相應的項目。最初,你有 W 資本。當你完成一個項目時,你將獲得純利潤,且利潤將被添加到你的總資本中。

總而言之,從給定項目中選擇最多 k 個不同項目的列表,以最大化最終資本,並輸出最終可獲得的最多資本。

示例 1:

輸入: k=2, W=0, Profits=[1,2,3], Capital=[0,1,1].

輸出: 4

解釋:
由於你的初始資本爲 0,你儘可以從 0 號項目開始。
在完成後,你將獲得 1 的利潤,你的總資本將變爲 1。
此時你可以選擇開始 1 號或 2 號項目。
由於你最多可以選擇兩個項目,所以你需要完成 2 號項目以獲得最大的資本。
因此,輸出最後最大化的資本,爲 0 + 1 + 3 = 4。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/ipo/

解析:

  • 一個小根堆存成本
  • 一個大根堆存利潤
  • 每次取成本小於資本的利潤放入大根堆,然後每次取大根堆裏的最大利潤(貪心)
  • 最終利潤即是答案
def findMaximizedCapital(k, W, Profits, Capital):
    from heapq import heapify, heappop, heappush
    H1, H2 = list(zip(Capital, Profits)), []
    heapify(H1)
    for _ in range(k):
        while H1 and H1[0][0] <= W:
            heappush(H2, -heappop(H1)[1])
        if not H2:
            break
        W -= heappop(H2)
    return W

2、最低加油次數

汽車從起點出發駛向目的地,該目的地位於出發位置東面 target 英里處。

沿途有加油站,每個 station[i] 代表一個加油站,它位於出發位置東面 station[i][0] 英里處,並且有 station[i][1] 升汽油。

假設汽車油箱的容量是無限的,其中最初有 startFuel 升燃料。它每行駛 1 英里就會用掉 1 升汽油。

當汽車到達加油站時,它可能停下來加油,將所有汽油從加油站轉移到汽車中。

爲了到達目的地,汽車所必要的最低加油次數是多少?如果無法到達目的地,則返回 -1 。

注意:如果汽車到達加油站時剩餘燃料爲 0,它仍然可以在那裏加油。如果汽車到達目的地時剩餘燃料爲 0,仍然認爲它已經到達目的地。

示例 1:

輸入:target = 1, startFuel = 1, stations = []
輸出:0
解釋:我們可以在不加油的情況下到達目的地。

示例 2:

輸入:target = 100, startFuel = 1, stations = [[10,100]]
輸出:-1
解釋:我們無法抵達目的地,甚至無法到達第一個加油站。

示例 3:

輸入:target = 100, startFuel = 10, stations = [[10,60],[20,30],[30,30],[60,40]]
輸出:2
解釋:
我們出發時有 10 升燃料。
我們開車來到距起點 10 英里處的加油站,消耗 10 升燃料。將汽油從 0 升加到 60 升。
然後,我們從 10 英里處的加油站開到 60 英里處的加油站(消耗 50 升燃料),
並將汽油從 10 升加到 50 升。然後我們開車抵達目的地。
我們沿途在1兩個加油站停靠,所以返回 2 。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/minimum-number-of-refueling-stops/

解析:

  • 每走一步將路過加油站的油存下來入堆,
  • 當油不夠時,將路過的加油站的最大油量加入當前油,繼續前進,
  • 當沒有存油時返回-1
  • 當達到目的地停止
def minRefuelStops(target, startFuel, stations):
    import heapq
    heap = []  # A maxheap is simulated using negative values
    stations.append((target, float('inf')))
    tank = startFuel
    ans = prev = 0
    for location, capacity in stations:
        tank -= location - prev
        while heap and tank < 0:  # must refuel in past
            tank += -heapq.heappop(heap)
            ans += 1
        if tank < 0:
            return -1
        heapq.heappush(heap, -capacity)
        prev = location

    return ans

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章