描述:
一个环形的公路上有n个加油站,编号为0,1,2,...n-1,
每个加油站加油都有一个上限,保存在列表limit中,即limit[i]为第i个加油站加油的上限,
而从第i个加油站开车开到第(i+1)%n个加油站需要cost[i]升油,cost为一个列表。
现在有一辆开始时没有油的车,要从一个加油站出发绕这个公路跑一圈回到起点。
给你整数n,列表limit和列表cost,你来判断能否完成任务。
如果能够完成任务,输出起始的加油站编号,如果有多个,输出编号最小的。
如果不能完成任务,输出-1。
解法一:
def judge(l):
length=len(l)
for i in range(0,length):
sum = l[i]
count = length
j=i
while count > 1:
if sum < 0:
break
j += 1
sum += l[j%length]
count -= 1
if count == 1:
return i
def oil(limit,cost):
if sum(limit) < sum(cost):
return -1
else:
l=map(lambda x,y:x-y,limit,cost)
return judge(l)
#limit = [2,1,1,1,4]
#cost = [1,3,2,2,1]
print oil(limit,cost)
解法二:
a = [(limit[i] - cost[i]) for i in range(n)]
z,times,ii = 0,0,0
for i in range(n<<1):
if times == n:
break
z += a[i%n]
if z < 0:
z,times,ii = 0,0,(i+1)%n
else:
times += 1
print ii if times==n else -1