代码随想录算法训练营之JAVA｜第三十天|134. 加油站

今天是第 天刷leetcode，立个flag，打卡60天，如果做不到，完成一件评论区点赞最高的挑战。

算法挑战链接

134. 加油站https://leetcode.cn/problems/gas-station/

第一想法

题目理解：汽车从A地到B地，可以在A地加油，在去B地的途中消耗油，要求求出从某地出发可以走完所有地方。

看到题目的时候的第一想法就是如何判断是否可以走完——只需要计算出这两个数组的差值，然后累加，如果小于0，则代表无论从哪个点出发都不行。

如果大于或者等于0，那么我们就要去寻找出发点。我使用的方法是将差值不小于0的都走一遍，于是代码如下：

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int dif[] = new int[gas.length];
        int addition = 0;
        for (int i = 0; i < gas.length; i++) {
            dif[i] = gas[i] - cost[i];
            addition += dif[i];
        }
        if (addition < 0) {
            return -1;
        }
        for (int i = 0; i < dif.length; i++) {
            if (dif[i] >= 0 && difStart(dif, i)) {
                return i;
            }
        }
        return -1;
    }

    private boolean difStart(int[] dif, int start) {
        int count = 0;
        int result = 0;
        for (int i = start; count < dif.length; i++) {
            int tmp = i % dif.length;
            result += dif[tmp];
            if (result < 0) {
                return false;
            }
            count++;
        }
        return true;
    }
}

 不出意外的没有AC，结果是超时。

超时的题目是前面超多0，最后面有个1，2. 晕死

于是脑袋一转，居然0这么特殊，那么我就特殊对待0就好了

于是有了下面的代码

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int dif[] = new int[gas.length];
        int addition = 0;
        for (int i = 0; i < gas.length; i++) {
            dif[i] = gas[i] - cost[i];
            addition += dif[i];
        }
        if (addition < 0) {
            return -1;
        }
        for (int i = 0; i < dif.length; i++) {
            if (dif[i] > 0 && difStart(dif, i)) {
                return i;
            }
        }
        for (int i = 0; i < dif.length; i++) {
            if (dif[i] == 0 && difStart(dif, i)) {
                return i;
            }
        }
        return -1;
    }

    private boolean difStart(int[] dif, int start) {
        int count = 0;
        int result = 0;
        for (int i = start; count < dif.length; i++) {
            int tmp = i % dif.length;
            result += dif[tmp];
            if (result < 0) {
                return false;
            }
            count++;
        }
        return true;
    }
}

看完代码随想录之后的想法 

正当我为我的小聪明沾沾自喜的时候，去看一下别人是怎么做的。结果大为震惊。

以下是摘抄：

贪心算法（方法二）

可以换一个思路，首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈，说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。

每个加油站的剩余量rest[i]为gas[i] - cost[i]。

i从0开始累加rest[i]，和记为curSum，一旦curSum小于零，说明[0, i]区间都不能作为起始位置，因为这个区间选择任何一个位置作为起点，到i这里都会断油，那么起始位置从i+1算起，再从0计算curSum。

如图：

那么为什么一旦[0，i] 区间和为负数，起始位置就可以是i+1呢，i+1后面就不会出现更大的负数？

如果出现更大的负数，就是更新i，那么起始位置又变成新的i+1了。

那有没有可能 [0，i] 区间 选某一个作为起点，累加到 i这里 curSum是不会小于零呢？ 如图：

如果 curSum<0 说明 区间和1 + 区间和2 < 0， 那么 假设从上图中的位置开始计数curSum不会小于0的话，就是 区间和2>0。

区间和1 + 区间和2 < 0 同时 区间和2>0，只能说明区间和1 < 0， 那么就会从假设的箭头初就开始从新选择其实位置了。

那么局部最优：当前累加rest[i]的和curSum一旦小于0，起始位置至少要是i+1，因为从i之前开始一定不行。全局最优：找到可以跑一圈的起始位置。

局部最优可以推出全局最优，找不出反例，试试贪心！

今日收获

当你得意的时候，往往就是你失败的时候。