题目:加油站
在一条环路上有 n
个加油站,其中第 i
个加油站有汽油 gas[i]
升。
你有一辆油箱容量无限的的汽车,从第 i
个加油站开往第 i+1
个加油站需要消耗汽油 cost[i]
升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas
和 cost
,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1
。如果存在解,则 保证 它是 唯一 的。
思路:如果总的汽油量大于总的消耗量,则一定有解,反之无解,题目又强调存在解,则保证它是唯一的,即唯一的出发点,从0出发点开始查询满足度,如果汽车行驶过程油量大于等于0,就可以一直继续,如果小于0,则小于0以及之前的位置都不能是出发点,则换下一个位置为出发点,汽车油量置空,重新出发,下1个位置遍历到最后一直有剩余油量,则下一个位置为出发点,遍历到最后的剩余油量一定够从0再到下一个位置之前的;并且剩余油量比下一个位置的以后位置的油量剩余之和也都大。
代码:
class Solution:
def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
if sum(gas) < sum(cost):
return -1
i = 0
start = 0
total_gas = 0
while i < len(gas):
total_gas += gas[i] - cost[i]
if total_gas < 0:
start = i + 1
total_gas = 0
i += 1
return start