Leetcode 134 题 加油站（标签：贪心算法）

题目描述

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

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

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

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

示例：
输入:
gas = [1,2,3,4,5]
cost = [3,4,5,1,2]
输出: 3
解释：从 3 号加油站出发，可以绕环路行驶一周。

思路分析

首先可以知道的是，如果总油量 sum(gas) 小于总消耗量 sum(cost)，那么肯定无法环行一周。

进一步分析，可以得到以下结论：

① 如果总油量 sum(gas) 大于等于总消耗量 sum(cost)，说明总油量可以把总消耗量填平，甚至还有余，那么肯定可以绕一周。

② 若汽车从 i 加油站出发恰好无法到达 j 加油站，那么 [i，j) 任意一点，均无法到达 j 加油站。

简单的画个图表示一下：

图中横坐标表示加油站位置，纵坐标表示汽车从 i 出发，剩余油量的变化，假设在 i 和 j 之间存在 k 加油站。

从 i 能正常到达 j，说明 i 到 k 时剩余油量必定大于等于0。即使有 i 到 k 这段路程的剩余油量加持，i 都无法到达 j，更何况少了这些剩余油量，k 是无论如何也到达不了 j 的。

以上两点可以推出，目标加油站必定在 [i, j) 之外。

代码描述

使用 Java 进行代码描述：

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int totalGas = 0;
        int totalCost = 0;

        for (int i = 0; i < gas.length; i++) {
            totalGas += gas[i];
            totalCost += cost[i];
        }

        if (totalGas < totalCost) { // 必定无法到达
            return -1;
        }

        int start = 0;  // 起始位置
        int sum = 0;    // 剩余油量
        for (int i = 0; i < gas.length; i++) {
            // gas[i]：当前加油站油量
            // cost[i]：到达下一个加油站需要消耗的油量
            sum += gas[i] - cost[i];
            if (sum < 0) {
                start = i + 1;
                sum = 0;
            }
        }

        return start;
    }
}

技术公众号：小猿君的算法笔记