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