python求小鱼游泳

Question

有一只小鱼，它平日每天游泳250公里，周末休息（实行双休日），假设从周X开始算起，过了N天以后，小鱼一共累计游泳了多少公里？
输入
3 10
输出
2000

Answer 1

x, n = [int(_) for _ in input().split()]

s = n * 250 - n // 7 * 2 * 250
if (n + x - 1) % 7 == 6:
    s -= 250

print(s)