题目:加油站
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;
}
};