LeetCode每日一题 - 134. 加油站（枚举+贪心）

题目：加油站

Description

在一条环路上有 N 个加油站，其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。
如果你可以绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1。

说明:

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

Solution

最直接的想法就是枚举每一个加油站作为起点，然后遍历一遍。然后TLE了。
加一个小优化就可以AC了。
枚举起点时，找到了一个不能通过的点。那这个点就是你下次枚举的起点。

AC Code

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int len = gas.size();
        for(int i=0;i<len;i++){
            int pos = i;
            int now=gas[pos];
            now-=cost[pos];
            pos++;
            while(now>=0){
                if(pos%len == i) return i;
                now+=gas[pos%len];
                now-=cost[pos%len];
                pos++;
            }
            i=pos -1;
        }
        return -1;
    }
};