class Solution:
# 动态规划
def minRefuelStops(self, target: int, startFuel: int, stations: List[List[int]]) -> int:
dp = [startFuel] + [0] * len(stations) # dp[i]表示加i次油可以行驶的最远距离
for i, (pos, gas) in enumerate(stations):
# 到达第i个加油站,则最多可以加油i次,每次减一来更新满足距离的最少加油次数
for t in range(i, -1, -1):
# 若存在加t次油可以到达当前pos,则可以更新到达此加油站多加一次油后能够行驶的最远距离
# 只有到达加油站,才能够加一些油行驶更远;若不能到达此加油站,则无法行驶更远
if dp[t] >= pos:
dp[t+1] = max(dp[t+1], dp[t] + gas)
for i in range(len(dp)):
if dp[i] >= target: # 第一次出现满足距离的,即需要最少次数,返回i
return i
return -1
class Solution:
# 优先队列
def minRefuelStops(self, target: int, startFuel: int, stations: List[List[int]]) -> int:
tank = startFuel # 油箱中的油量
res = 0
prev = 0 # 上一个加油站的位置
pq = [] # 大顶堆(python默认为小顶堆,需要取反)
stations.append((target, float('inf')))
for location, gas in stations:
tank -= location - prev # 每路过一个加油站,油箱中则消耗掉这两个加油站之间的路程的油量
# 若当前油量不够到达当前加油站,且前面有加油站,则在油量最多的加油站加油
while pq and tank < 0:
tank += -heapq.heappop(pq) # 小顶堆顶弹出最小值(取反得到最大值)
res += 1 # 加油次数则加一
if tank < 0: return -1 # 若加了前面所有加油站的油都无法到达,则返回-1
heapq.heappush(pq, -gas) # 当前油量入堆
prev = location # 更新上一个加油站的位置
return res
134. 加油站
class Solution:
def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
# 若总加油量低于耗油量,则不可达
if sum(gas) < sum(cost):
return -1
n = len(gas)
res = 0
cur_sum = 0
for i in range(n):
# 某一个加油站加的油与耗油量做差,剩余的油量加在油箱中
cur_sum += gas[i] - cost[i]
# 若当前剩余的油量为负,则当前点不可达
# 把下一个点作为开始点,
if cur_sum < 0:
cur_sum = 0
res = i + 1
return res
- 解法2