美文网首页
LeetCode 力扣 134. 加油站

LeetCode 力扣 134. 加油站

作者: windliang | 来源:发表于2020-04-27 10:41 被阅读0次

    题目描述(中等难度)

    把这个题理解成下边的图就可以。

    每个节点表示添加的油量,每条边表示消耗的油量。题目的意思就是问我们从哪个节点出发,还可以回到该节点。只能顺时针方向走。

    解法一 暴力解法

    考虑暴力破解,一方面是验证下自己对题目的理解是否正确,另一方面后续的优化也可以从这里入手。

    考虑从第 0 个点出发,能否回到第 0 个点。

    考虑从第 1 个点出发,能否回到第 1 个点。

    考虑从第 2 个点出发,能否回到第 2 个点。

    ... ...

    考虑从第 n 个点出发,能否回到第 n 个点。

    由于是个圆,得到下一个点的时候我们需要取余数。

    public int canCompleteCircuit(int[] gas, int[] cost) {
        int n = gas.length;
        //考虑从每一个点出发
        for (int i = 0; i < n; i++) {
            int j = i;
            int remain = gas[i];
            //当前剩余的油能否到达下一个点
            while (remain - cost[j] >= 0) {
                //减去花费的加上新的点的补给
                remain = remain - cost[j] + gas[(j + 1) % n];
                j = (j + 1) % n;
                //j 回到了 i
                if (j == i) {
                    return i;
                }
            }
        }
        //任何点都不可以
        return -1;
    }
    

    解法二 优化尝试一

    暴力破解慢的原因就是会进行很多重复的计算。比如下边的情况:

    假设当前在考虑 i,先初始化 j = i
    * * * * * *
          ^
          i
          ^
          j
          
    随后 j 会进行后移
    * * * * * *
          ^ ^
          i j
          
    继续后移
    * * * * * *
          ^   ^
          i   j
          
    继续后移
    * * * * * *
    ^     ^   
    j     i   
    
    此时 j 又回到了第 0 个位置,我们在之前已经考虑过了这个位置。
    如果之前考虑第 0 个位置的时候,最远到了第 2 个位置。
    那么此时 j 就可以直接跳到第 2 个位置,同时加上当时的剩余汽油,继续考虑
    * * * * * *
        ^ ^   
        j i  
    

    利用上边的思想我们可以进行一个优化,就是每考虑一个点,就将当前点能够到达的最远距离记录下来,同时到达最远距离时候的剩余汽油也要记下来。

    public int canCompleteCircuit(int[] gas, int[] cost) {
        int n = gas.length;
        //记录能到的最远距离
        int[] farIndex = new int[n];
        for (int i = 0; i < farIndex.length; i++) {
            farIndex[i] = -1;
        }
        //记录到达最远距离时候剩余的汽油
        int[] farIndexRemain = new int[n];
        for (int i = 0; i < n; i++) {
            int j = i;
            int remain = gas[i];
            while (remain - cost[j] >= 0) {
                //到达下个点后的剩余
                remain = remain - cost[j];
                j = (j + 1) % n;
                //判断之前有没有考虑过这个点
                if (farIndex[j] != -1) {
                    //加上当时剩余的汽油
                    remain = remain + farIndexRemain[j];
                    //j 进行跳跃
                    j = farIndex[j];
                } else {
                    //加上当前点的补给
                    remain = remain + gas[j];
                }
                if (j == i) {
                    return i;
                }
            }
            //记录当前点最远到达哪里
            farIndex[i] = j;
            //记录当前点的剩余
            farIndexRemain[i] = remain;
        }
        return -1;
    
    }
    

    遗憾的是,这个想法针对 leetcode 的测试集速度上没有带来很明显的提升。不过记录已经求出来的解进行优化,这个思想还是经常用的,也就是空间换时间。

    让我们换个思路继续优化。

    解法三 优化尝试二

    我们考虑一下下边的情况。

    * * * * * *
    ^     ^
    i     j
    

    当考虑 i 能到达的最远的时候,假设是 j

    那么 i + 1j 之间的节点是不是就都不可能绕一圈了?

    假设 i + 1 的节点能绕一圈,那么就意味着从 i + 1 开始一定能到达 j + 1

    又因为从 i 能到达 i + 1,所以从 i 也能到达 j + 1

    但事实上,i 最远到达 j 。产生矛盾,所以 i + 1 的节点一定不能绕一圈。同理,其他的也是一样的证明。

    所以下一次的 i 我们不需要从 i + 1 开始考虑,直接从 j + 1 开始考虑即可。

    还有一种情况,就是因为到达末尾的时候,会回到 0

    如果对于下边的情况。

    * * * * * *
      ^   ^
      j   i
    

    如果 i 最远能够到达 j ,根据上边的结论 i + 1j 之间的节点都不可能绕一圈了。想象成一个圆,所以 i 后边的节点就都不需要考虑了,直接返回 -1 即可。

    public int canCompleteCircuit(int[] gas, int[] cost) {
        int n = gas.length;
        for (int i = 0; i < n; i++) {
            int j = i;
            int remain = gas[i];
            while (remain - cost[j] >= 0) {
                //减去花费的加上新的点的补给
                remain = remain - cost[j] + gas[(j + 1) % n];
                j = (j + 1) % n;
                //j 回到了 i
                if (j == i) {
                    return i;
                }
            }
            //最远距离绕到了之前,所以 i 后边的都不可能绕一圈了
            if (j < i) {
                return -1;
            }
            //i 直接跳到 j,外层 for 循环执行 i++,相当于从 j + 1 开始考虑
            i = j;
    
        }
        return -1;
    }
    

    写题的时候先写出暴力的解法,然后再考虑优化,有时候是一种不错的选择。

    更多详细通俗题解详见 leetcode.wang

    相关文章

      网友评论

          本文标题:LeetCode 力扣 134. 加油站

          本文链接:https://www.haomeiwen.com/subject/zhxywhtx.html