Gas Station [leetcode] 的两种解法

最新推荐文章于 2017-12-06 08:35:23 发布

原创最新推荐文章于 2017-12-06 08:35:23 发布 · 923 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种解决环形路线中车辆是否能完成一圈旅行的问题，并提供了两种算法实现方式。第一种通过寻找最小剩余油量的位置来确定起点；第二种在遇到油量不足时重新设置起点，直至找到可行方案。

由于gas总量大于cost总量时，一定可以绕所有城市一圈。

第一种解法：

假设一开始有足够的油，从位置i出发，到位置k时剩余的油量为L(i,k)。

对任意的k，L(i,k)根据i的不同，只相差常数。

我们只需要找到最小的L(0, k)对应的k，k+1为所求。

代码如下：

    int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
        int start = 0;
        int curGas = 0, minGas = 0, totalGas = 0;
        for (int i = 0; i < gas.size(); i++)
        {
            int temp = gas[i] - cost[i];
            curGas += temp;
            totalGas += temp;
            if (minGas > curGas)
            {
                start = i + 1;
                minGas = curGas;
            }
        }
        if (totalGas >= 0) return start % gas.size();
        else return -1;
    }

第二种解法：

如果L(i,k) < 0，则从i和k之间所有的位置都不能到k

所以从k+1的位置从0开始找

    int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
        int start = 0;
        int curGas = 0, totalGas = 0;
        for (int i = 0; i < gas.size(); i++)
        {
            int temp = gas[i] - cost[i];
            curGas += temp;
            totalGas += temp;
            if (curGas < 0)
            {
                start = i + 1;
                curGas = 0;
            }
        }
        if (totalGas >= 0) return start % gas.size();
        else return -1;
    }