LintCode : 加油站

最新推荐文章于 2022-02-14 14:08:09 发布

orekinana

最新推荐文章于 2022-02-14 14:08:09 发布

阅读量360

点赞数

分类专栏： LintCode 文章标签：贪心

本文链接：https://blog.csdn.net/orekinana/article/details/64443898

版权

LintCode 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

$#LintCode ：加油站

题目

在一条环路上有 N 个加油站，其中第 i 个加油站有汽油gas[i]，并且从第i个加油站前往第i+1个加油站需要消耗汽油cost[i]。
你有一辆油箱容量无限大的汽车，现在要从某一个加油站出发绕环路一周，一开始油箱为空。
求可环绕环路一周时出发的加油站的编号，若不存在环绕一周的方案，则返回-1。

样例

现在有4个加油站，汽油量gas[i]=[1, 1, 3, 1]，环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]。则出发的加油站的编号为2。

思路

最暴力的 $O(n^2)$ 的算法就是从起点开始检查是否可以绕一圈，碰到不足以走到下一个加油站的则从起点+1继续检查，直到检查完n个加油站。

当然这题是有O(n)的解法的，不用那么暴力。我们知道，从起点start开始走，当我们走到第i个加油站发现不足以走到i+1个加油站时我们不需要从start+1开始查找，因为既然从start开始到不了i+1那么从start到i中间的任意一节点开始都是到不了i+1的，所以下一个我们就可以从i+1开始检查，直到发现满足条件的。

代码

int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
    int start = 0;
    while(start != gas.size()) {
        int total = 0;
        int firsttime = 1;
        int once = 0;
        for(int i = start; i < gas.size(); i++) {
            if(i == start && firsttime == 0) 
                return start;
            total += gas[i] - cost[i];
            if(total < 0) {
                start = i + 1;
                break;
            }
            if(i == gas.size() - 1) {
                i = -1;
                once = 1;
            } 
            firsttime = 0;
        }
        if(once == 1)
            break;
    }
    return -1;
}