LeetCode 134. 加油站 | Python

134. 加油站

---

题目来源：力扣（LeetCode）https://leetcode-cn.com/problems/gas-station/

题目

---

在一条环路上有 N 个加油站，其中第 i 个加油站有汽油 gas[i] 升。

你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。

如果你可以绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1。

说明:

• 如果题目有解，该答案即为唯一答案。
• 输入数组均为非空数组，且长度相同。
• 输入数组中的元素均为非负数。

示例 1:

输入: 
gas  = [1,2,3,4,5]
cost = [3,4,5,1,2]

输出: 3

解释:
从 3 号加油站(索引为 3 处)出发，可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站，此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站，此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站，此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站，此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站，你需要消耗 5 升汽油，正好足够你返回到 3 号加油站。
因此，3 可为起始索引。

示例 2:

输入: 
gas  = [2,3,4]
cost = [3,4,3]

输出: -1

解释:
你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发，可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站，此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站，此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站，因为返程需要消耗 4 升汽油，但是你的油箱只有 3 升汽油。
因此，无论怎样，你都不可能绕环路行驶一周。

解题思路

---

思路：贪心算法

先审题，题目中给出一条环路，环路上有 $N$ 个加油站，其中第 $i$ 个加油站有 $gas[i]$ 升汽油。而从第 $i$ 个加油站开往第 $i+1$ 个加油站时，需要消耗 $cost[i]$ 升汽油。

假设现在有一辆油箱容量无限的汽车，开始油箱为空，问是否存在从其中一个加油站出发，能够绕环路行驶一周？如果存在，返回出发加油站的编号；否则，返回 $-1$。

其中题目说明：

• 题目有解时，仅有唯一解；
• 数组非空，且元素均为非负数。

这里最直观的解法，就是遍历每个加油站，检查是否能够绕环路行驶一周？但这里会存在重复检查的现象。

现在，假设从第 $i$ 个加油站出发，能够到达第 $j$ 个加油站，但是剩余汽油无法支撑到下一个加油站。问第 $i$ 个加油站到第 $j$ 个加油站之间，其中任意一个加油站作为出发点，是否能够到达第 $j$ 个加油站的下一个加油站？

假设从第 $i$ 个加油站出发，能够到达第 $j$ 个加油站。这里能够第 $j$ 个加油站，也就是说前面从加油站 $i$ 出发到第 $j$ 个加油站汽油充足，只是第 $j$ 个加油站的提供的汽油，与前面所剩余的汽油合计无法支撑到达下一个加油站。所以在加油站 $i$ 到加油站 $j$ 之间任意一个加油站 $k$，从 $i$ 出发，都能到达 $k$。那么此时：
$$式一：\sum _{x=i}^{k}gas[x]\ge \sum _{x=i}^{k}cost[x]（其中，k\in [i,j)）$$

$$式二：\sum _{x=i}^{j}gas[x]<\sum _{x=i}^{j}cost[x]$$

其中：

• 式一，表示在加油站 $i$ 到加油站 $j$ 之间任意一个加油站 $k$，从 $i$ 出发，都能到达 $k$；
• 式二，表示从第 $i$ 个加油站出发，能够到达第 $j$ 个加油站，但是剩余汽油无法支撑到下一个加油站。

那么现在考虑，第 $i$ 个加油站到第 $j$ 个加油站之间，是否存在某个加油站作为起始点，能够到达第 $j$ 个加油站的下一个加油站？

假设存在加油站 $h$（其中 $h\in [i,j]$），现在要求：
$$\sum _{x=h}^{j}gas[x]?\sum _{x=h}^{j}cost[x]$$
由式一、式二，可得：
$$\begin{array}{rl}\sum _{x=h}^{j}gas[x]& =\sum _{x=i}^{j}gas[x]-\sum _{x=i}^{h-1}gas[x]\\ & <\sum _{x=i}^{j}cost[x]-\sum _{x=i}^{h-1}gas[x](由式二得)\\ & <\sum _{x=i}^{j}cost[x]-\sum _{x=i}^{h-1}cost[x]=\sum _{x=h}^{j}cost[x](由式一得)\end{array}$$
即是：
$$\sum _{x=h}^{j}gas[x]<\sum _{x=h}^{j}cost[x]$$

所以当加油站 $i$ 出发能够到达加油站 $j$，但是无法到达下一个加油站时。从加油站 $i$ 到加油站 $j$ 中任取一个加油站也无法到达下一个加油站。

那么这里就可以直接取加油站 $j$ 的下一个加油站作为起点，再次检查。

具体的代码实现如下。

class Solution:
    def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
        n = len(gas)
        # 先从第一个加油站开始遍历
        i = 0
        while i < n:
            # 初始可获得的汽油及消耗汽油均为 0
            sum_of_gas = 0
            sum_of_cost = 0
            # 开始检查是否能够走完一圈
            count = 0
            while count < n:
                # 因为道路是环形的，注意索引
                j = i + count if i + count < n else i + count - n
                # 统计可获得汽油量及消耗量总和，进行比较
                sum_of_gas += gas[j]
                sum_of_cost += cost[j]
                # 获得汽油量小于消耗量时，表示汽油无法支撑到下一个加油站
                if sum_of_gas < sum_of_cost:
                    break
                # 若油量支持驶向下个站，则继续检查
                count += 1
            
            # 如果 count 等于 n，表示走完所有的加油站，
            # 那么返回起始加油站编号 i
            if count == n:
                return i
            # 否则起始点将重置为最后能到达的加油站的下一个加油站，
            # 其中 i + count 表示能到达的最后一个加油站
            else:
                i = i + count + 1
        # 若均无法走完一圈，返回 -1
        return -1

欢迎关注

---

公众号 【书所集录】

---

如有错误，烦请指出，欢迎指点交流。