leetcode-871. 最低加油次数

题目

汽车从起点出发驶向目的地，该目的地位于出发位置东面 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

解题思路

堆的应用。

加油的过程，可以看成当没油时，从已经走过的加油站里取油加进去。为了加油次数最少，应该在没油的时候，从走过的加油站里挑油量最大的加油站加油。

用大根堆保存走过的加油站的油量，当发现没油时，挑出来最大的油量加入，然后继续走。如果发现大根堆空了（油全部加完了）也没有走到终点，则说明走不到了，此时返回即可。

Time complexity: $o(n\log n)$
Space complexity: $o(n)$

代码

class Solution:
    def minRefuelStops(self, target: int, startFuel: int, stations: List[List[int]]) -> int:
        fuel_heap = []
        max_mile = startFuel
        refuel_cnt = 0
        for station_loc, station_fuel in stations:
            while target > max_mile and fuel_heap and station_loc > max_mile:
                max_mile -= heapq.heappop(fuel_heap)
                refuel_cnt += 1
            if target <= max_mile:
                break
            elif station_loc <= max_mile:
                heapq.heappush(fuel_heap, -station_fuel)
            else:
                return -1
        while fuel_heap and max_mile < target:
            max_mile -= heapq.heappop(fuel_heap)
            refuel_cnt += 1
        return refuel_cnt if max_mile >= target else -1