描述
环形路上有n个加油站,第i个加油站的汽油量是gas[i].
你有一辆车,车的油箱可以无限装汽油。从加油站i走到下一个加油站(i+1)花费的油量是cost[i],你从一个加油站出发,刚开始的时候油箱里面没有汽油。
求从哪个加油站出发可以在环形路上走一圈。返回加油站的下标,如果没有答案的话返回-1。
注意:
答案保证唯一。
示例1
输入:
[2,3,1],[3,1,2]
返回值:
1
//复杂度O(n)
class Solution {
public:
int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
//total记录全局是否有解,sum记录i到i+1是否可行
int total = 0, sum = 0;
int index = -1;
for (int i = 0; i < gas.size(); i++) {
sum += gas[i] - cost[i]; //本次消耗
total += gas[i] - cost[i]; //总消耗
if (sum < 0) {
sum = 0;
index = i; //记录解的位置
}
}
return total >= 0 ? index + 1 : -1;//只要total>=0,肯定有解
}
};