python自学做题记录之加油站

题目描述：
一个环形的公路上有n个加油站，编号为0,1,2,…n-1, 每个加油站加油都有一个上限，保存在列表limit中，即limit[i]为第i个加油站加油的上限， 而从第i个加油站开车开到第(i+1)%n个加油站需要cost[i]升油,cost为一个列表。 现在有一辆开始时没有油的车，要从一个加油站出发绕这个公路跑一圈回到起点。 给你整数n，列表limit和列表cost,你来判断能否完成任务。 如果能够完成任务，输出起始的加油站编号，如果有多个,输出编号最小的。 如果不能完成任务，输出-1。
示例：
输入：n = 2 limit = [1, 2] cost = [2, 2]

输出：-1

分析一下
首先这个公路是环形，所以要用到%n
所以我们就可以用两个循环来实现这个算法
第一层num是控制起始的加油站编号
第二层i是控制加油次数，一共要加n次，可以直接0~n-1
然后每次就判断现有的油量和所需的油量进行对比，如果少了，直接退出第二层循环，开始下一个加油站编号
如果多了，要记得减掉消耗的，还有就是注意加一个i!=n-1最后一次比较不需要减油
如果油还多并且是最后一站，就输出，判断条件f=1
还需要注意多解的情况，所以如果出现了符合条件的编号，直接两个break结束循环，因为只需要最小的一个编号即可
下面是代码

oil=0
f=0
for num in range(n):
    for i in range(n):
        oil=oil+limit[(num+i)%n]
        if oil<cost[(num+i)%n]:
            oil=0
            break
        if oil>=cost[(num+i)%n] and i!=n-1:
            oil=oil-cost[(num+i)%n]
        if oil>=cost[(num+i)%n] and i==n-1:
            print(num)
            f=1
            break
	if f==1:
		break
if f==0:
    print(-1)