小鱼游泳距离计算-python

题目描述
有一只小鱼，它平日每天游泳 250 公里，周末休息（实行双休日)，假设从周 x(1≤x≤7) 开始算起，过了 n(n≤106) 天以后，小鱼一共累计游泳了多少公里呢？

输入格式
输入两个整数x,n(表示从周x算起，经过n天）。

输出格式
输出一个整数，表示小鱼累计游泳了多少公里。

输入样例：3,10

输出样例：2000

x = int(input('小鱼从第周几开始游泳(1-7)：'))
n = int(input('小鱼游泳天数：'))

#定义初始距离
distance = 0
#定义移动距离列表
list1 = [250, 250, 250, 250, 250, 0, 0]

for i in range(x-1,x+n):
    #新的一周重头继续叠加计算
    if i//7>=1:
        i = i%7
        distance = list1[i] + distance
    else:
        distance = list1[i] + distance

print(distance)