加油站
在一条环路上有 n
个加油站,其中第 i
个加油站有汽油 gas[i]
升。
你有一辆油箱容量无限的的汽车,从第 i
个加油站开往第 i+1
个加油站需要消耗汽油 cost[i]
升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas
和 cost
,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -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 升汽油。 因此,无论怎样,你都不可能绕环路行驶一周。
提示:
gas.length == n
cost.length == n
1 <= n <= 105
0 <= gas[i], cost[i] <= 104
解题思路
首先我们需要确定作为起点的加油站,假设是x,需要满足条件:当前第x个加油站的汽油gas[x]需要大于或者等于达到第x个加油站所消耗的汽油cost[x]。
所以,我们可以从第一个加油站开始遍历,累计加油站的汽油减去累计消耗的汽油,检查是否大于等于0;如果满足条件,就继续到下一个加油站,并继续检查是否满足条件。
不能作为起点的第y个加油站
如果到的加油站不满足条件,说明从起点到第y个加油站之间的任何一个加油站都是不能作为起点的,我们可以推导一下这个结论:
假设起点为第x个加油站,最后一个可以到达的加油站是第y个,第z个加油站为起点到第y个加油站之前的任何一个加油站,我们可以得到如下条件:
- 条件1:0 <= x < z <= y - 1 < y <= n - 1
- 条件2:起点的汽油大于或者等于消耗的汽油,即为gas[x] >= cost[x]
- 条件3:,累计的汽油应该是大于或者等于累计消耗的汽油,即为gas[x] + ... + gas[z] >= cost[x] + ... + cost[z]
- 条件4:起点到当前第y个加油站,累计的汽油是小于累计消耗的汽油,即为gas[x] + ... + gas[y] < cost[x] + ... + cost[y]
现在,我们考虑一下在x和y之间的任何一个加油站z,我们从加油站z出发,看能否达到y的下一个加油站,也就是gas[z] + ... + gas[y] 是否大于或者等于 cost[z] + ... + cost[y]。
我们继续推导:
gas[z] + ... + gas[y]
= (gas[x] + ... + gas[y]) - (gas[x] + ... + gas[z-1])
< (cost[x] + ... + cost[y]) - (gas[x] + ... + gas[z-1])
< (cost[x] + ... + cost[y]) - (cost[x] + ... + cost[z-1])
= cost[z] + ... + cost[y]
所以,gas[z] + ... + gas[y] < cost[z] + ... + cost[y],即在x和y之间的任何一个加油站z到达y以后,累计的汽油小于累计消耗的汽油,所以无法到达y的下一个加油站。
也就是说当到达y以后,发现累计的汽油小于累计消耗的汽油时,我们需要用y以后的加油站作为起点,并继续遍历;只有当累计加油站的汽油大于或者等于累计消耗的汽油时,才有起点,否则不存在起点。
可以作为起点的第y+1个加油站
假设第y+1个加油站可以作为起点,根据上面的推导,我们可以得到如下条件:
- 条件5:gas[0] + ... + gas[y-1] >= cost[0] + ... + cost[y-1],由条件1可以推导出这个公式,并且在第一个加油站和第y-1加油站之间的任何一个加油站,都是符合条件5的
- 条件6:gas[y] < cost[y]
- 条件7:gas[y+1] + ... + gas[n-1] >= cost[y+1] + ... + cost[n-1],并且从第y+1个到第n-1个之间任何一个加油站,都是符合条件7的
- 条件8:gas[0] + ... + gas[n-1] >= cost[0] + ... + cost[n-1],即全程累计的汽油是足够的
所以,由条件7和条件5,我们可以推导出,从第y+1个加油站,到第n-1个加油站,然后继续到第y-1个加油站,都是满足条件的,而到第y个加油站就是全程了。
具体代码如下:
class Solution {
public int canCompleteCircuit(int[] gas, int[] cost) {
int curr_tank = 0;
int total_tank = 0;
int start_station = 0;
for(int i = 0; i < gas.length; i++){
total_tank += gas[i] - cost[i];
curr_tank += gas[i] - cost[i];
if(curr_tank < 0){
curr_tank = 0;
start_station = i + 1;
}
}
return total_tank >= 0 ? start_station : -1;
}
}
复杂度分析
- 时间复杂度:O(n),需要遍历一遍数组。
- 空间复杂度:O(1),只需要常量的空间存储变量。