在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i]
,并且从第_i_个加油站前往第_i_+1个加油站需要消耗汽油cost[i]
。
你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。
求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1
。
Yes
样例
现在有4个加油站,汽油量gas[i]=[1, 1, 3, 1]
,环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]
。则出发的加油站的编号为2。
注意
数据保证答案唯一。
挑战
标签
Expand
O(n)时间和O(1)额外空间
分析:其实只要总的汽油量要大于总的消耗量,那么肯定是有解的,可以从头遍历起,什么时候汽油量小于消耗量了,就假设从下一个点重新开始。
代码:
class Solution {
public:
/**
* @param gas: a vector of integers
* @param cost: a vector of integers
* @return: an integer
*/
int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
// write your code here
int cur = 0;
int totalGas = 0;
int totalCost = 0;
int start = 0;
for(int i=0;i<gas.size();i++)
{
cur+=gas[i];
cur-=cost[i];
totalGas+=gas[i];
totalCost+=cost[i];
if(cur<0)
{
start = i+1;
cur = 0;
}
}
if(totalGas>=totalCost)
return start;
return -1;
}
};