首页 > 代码库 > Leetcode 134 Gas Station

Leetcode 134 Gas Station

There are N gas stations along a circular route, where the amount of gas at station i is gas[i].

You have a car with an unlimited gas tank and it costs cost[i] of gas to travel from station i to its next station (i+1). You begin the journey with an empty tank at one of the gas stations.

Return the starting gas station‘s index if you can travel around the circuit once, otherwise return -1.

Note:
The solution is guaranteed to be unique.

 

这道题最容易想到的方法就是从一个点出发,看看能不能走回到起点,如果可以返回起点的index, 不行的话再试下一个点。但是超时。

这个超时算法中有两点值得注意。

a) L13. 对于 int a 怎么计算 a 在 循环群[0,1,2,n-1]上的index?

b) L21-L22,可以使用一个变量记录是哪一步跳出的。然后判断是不是最后一步跳出的。

 1 class Solution(object):
 2     def canCompleteCircuit(self, gas, cost):
 3         """
 4         :type gas: List[int]
 5         :type cost: List[int]
 6         :rtype: int
 7         """
 8         n = len(gas)
 9         for i in range(n):
10             fule = 0
11             step = 0
12             for j in range(n):
13                 index = (i+j +1)%n -1
14                 fule += gas[index]
15                 step = j
16                 if fule < cost[index]:
17                     break
18                 else:
19                     fule -= cost[index]
20 
21                 if step == n-1:                        
22                     return i
23         return -1

 

一下三条原则摘自: http://bangbingsyb.blogspot.com/2014/11/leetcode-gas-station.html

性质1. 对于任意一个加油站i,假如从i出发可以环绕一圈,则i一定可以到达任何一个加油站。显而易见。

 
性质2. 如果这样的i是唯一的,则必然不存在j!=i, 从j出发可以到达i。
 
反证法:如果存在这样的j,则必然存在j->i->i的路径,而这个路径会覆盖j->j一周的路径。那么j也将是一个符合条件的起点。与唯一解的限制条件矛盾。
 
性质3. 假如i是最后的解,则由1可知,从0 ~ i-1出发无法达到i。而从i出发可以到达i+1 ~ n-1。
 
 1 class Solution(object):
 2     def canCompleteCircuit(self, gas, cost):
 3         """
 4         :type gas: List[int]
 5         :type cost: List[int]
 6         :rtype: int
 7         """
 8         n = len(gas)
 9         start = 0
10         netGasSum = 0
11         curGasSum = 0
12         
13         for i in range(n):
14             netGasSum += gas[i] - cost[i]
15             curGasSum += gas[i] - cost[i]
16             if curGasSum < 0:
17                 start = i+1
18                 curGasSum = 0;
19         
20         if netGasSum < 0:
21             return -1
22         return start

 

 

Leetcode 134 Gas Station