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;
    }
};

猜你喜欢

转载自blog.csdn.net/buibuilili/article/details/109791480