在一条环路上有 N 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
如果你可以绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1。
说明:
如果题目有解,该答案即为唯一答案。
输入数组均为非空数组,且长度相同。
输入数组中的元素均为非负数。
示例 1:
输入:
gas = [1,2,3,4,5]
cost = [3,4,5,1,2]
输出: 3
解释:
从 3 号加油站(索引为 3 处)出发,可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站,此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站,此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站,此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站,此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站,你需要消耗 5 升汽油,正好足够你返回到 3 号加油站。
因此,3 可为起始索引。
示例 2:
输入:
gas = [2,3,4]
cost = [3,4,3]
输出: -1
解释:
你不能从 0 号或 1 号加油站出发,因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发,可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站,此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站,此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站,因为返程需要消耗 4 升汽油,但是你的油箱只有 3 升汽油。
因此,无论怎样,你都不可能绕环路行驶一周。
解析:
首先因为初始油箱为空,因此可以得到一个简单的结论,如果全程加的油少于消耗的油,则必然不可能行驶一周。反之则必然可以行驶一周。
假设全程加的油等于路程消耗的油。如果每一个加油站都无法到达下一个,即g[n]-c[n]<0 ,那么总消耗将会大于总输入,与前提不符。因此必然有一个加油站可以到达下一个加油站,且g[n]-c[n] = -(g[…]-c[…]),即额外获取的油可以弥补其他加油站的消耗,完成整个路程。
首先计算全程的总消耗与总输入,保证必定可以行驶一周。
我们从第一个加油站开始,即从下标0开始,定义nowgas代表油箱内的油,初始为0
定义res代表起始点,初始为0
对于每一个加油站,我们计算它的g[n]-c[n],累加入nowgas
如果nowgas小于0,则代表从res起点开始,无法到达第n+1个加油站,即这一段的总消耗大于总输入,于是将nowgas清零,并将res定义为n+1,即从下一个加油站重新开始;
如果nowgas大于0,则代表从res起点开始,能够到达第n+1个加油站,那么一切不变,继续进行遍历。
遍历完所有的加油站,可以得到目前的res起始点,那么可以得到如下结论:我们可以从res点开始,到达最后一个加油站并绕回第一个加油站。
那么剩下的问题在于是否能够在这个基础上从第一个加油站返回res起始点。
设从res到达第一个加油站总输入为a,从第一个加油站返回res总输入为b
从res到达第一个加油站总消耗为c,从第一个加油站返回res总消耗为d
根据大前提得出a+b-c-d>=0
因为从res能够到达第一个加油站,因此a-c>=0
如果无法从在此基础上从第一个加油站返回res,则(a-c)+(b-d) <0
与a+b-c-d>=0不符,
因此必然可以从res点绕行一周返回res点。
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
if(accumulate(gas.begin(),gas.end(),0)<accumulate(cost.begin(),cost.end(),0)) return -1;
int nowgas = 0;
int res = 0;
for(int i=0;i<gas.size();++i){
nowgas += gas[i] - cost[i];
if(nowgas<0){
res = i+1;
nowgas = 0;
}
}
return res;
}
};