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

思路

将本题看成是修路问题，最开始有一段路，路中间有补充的路（这里简称补充站），要修到指定长度的路。每路过一个补充站，我们都保留这个补充站里补休的路长，当目前的路长到不了下一个补充站时，我们考虑从已经保留的补休路长拿出来几段补休，每次拿出来最长的一段补休到当前路段上，循环上述过程，直到当前路长等于题中要求的指定长度，我们从补充站中拿出补休路段数就是答案。
具体解题可看下方代码注释

题解

class Solution:
    def minRefuelStops(self, target: int, startFuel: int, stations: List[List[int]]) -> int:
        # 特殊情况直接返回
        # 不用加油
        if startFuel >= target:
            return 0
        # 第一个加油站都到不了，或者没有加油站
        if len(stations) == 0 or startFuel < stations[0][0]:
            return -1
        # 加油集合，加油次数
        dp, index = [], 0
        # 跑加油站过程
        for i in range(len(stations)):
            # 当前油量够，直接在加油集合中添加油量
            if startFuel >= stations[i][0]:
                dp.append(stations[i][1])
            else:
                # 油量不够，从加油集合中每次取最大的加油量，直到能到下一个加油站的油量
                while dp:
                    if startFuel < stations[i][0]:
                        startFuel += max(dp)
                        dp.remove(max(dp))
                        index += 1
                    else:
                        break
               # 加油集合中加入当前加油站的油量
                if startFuel >= stations[i][0]:
                    dp.append(stations[i][1])
                # 如果加油集合中加完还不够，说明不能再往下进行直接返回
                else:
                    return -1
        # 上面是加油站跑完的过程，下面还需判断是否能跑完全程
        # 最后判断油量是否能跑完全程，不能的话继续从加油集合中添加
        while dp:
            if startFuel < target:
                startFuel += max(dp)
                dp.remove(max(dp))
                index += 1
            else:
                break
        # 能跑完全程，返回次数
        if startFuel >= target:
            return index
        # 所有油都加完也跑不完的情况
        else:
            return -1