题目链接
题目描述
汽车从起点出发驶向目的地,该目的地位于出发位置东面
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 。提示:
1 <= target, startFuel, stations[i][1] <= 10^9
0 <= stations.length <= 500
0 < stations[0][0] < stations[1][0] < ... < stations[stations.length-1][0] < target
解题思路
贪心策略+优先队列(大顶堆)
每次路过加油站先不加油,将油量存入大顶堆,当油量不足以走到下个加油站时,选择堆中最多的油站加油,直到车可以走到下个加油站或目的地,如果堆为空仍无法走到下个加油站或目的地,则返回-1
代码
Python
import heapq
class Solution:
def minRefuelStops(self, target: int, startFuel: int, stations: list[list[int]]) -> int:
'''
cur:当前油量能达到的最远距离
q:大顶堆(因为python中的heapq的堆是小顶堆,因此需要用负数存储
ans:加油次数
'''
cur, q, ans = startFuel, [], 0
for pos, fuel in stations+ [[target, 0]]: #这里加[[target, 0]]是把终点看作最后一个加油站,而且油量为0
while cur < pos: #加油条件:车的油量不足以到达下个加油站时,需要从大顶堆弹出一个最大油量的加油站进行加油
if not q: return -1 #堆中无油,代表车辆无法到达下个加油站
cur += -heapq.heappop(q)
ans += 1
heapq.heappush(q, -fuel) #不需要加油,把油量存入堆中
return ans