CSDN编程竞赛第三期小鱼的航程(改进版)

题目：

小鱼的航程(改进版)

有一只小鱼，它上午游泳150公里，下午游泳100公里，晚上和周末都休息（实行双休日)，假设从周x(1<=x<=7)开始算起，请问这样过了n天以后，小鱼一共累计游泳了多少公里呢？
例：周三开始游泳，10天后，共2000公里

---

描述

周末都休息，实行双休日，则周六、周日停止

---

解决方案

python

#k表示开始周几
#n表示经过多少天
#一个星期7天，n天，看下这n天有多少周
def fun(k, n):
	week = n//7
	#看下剩余天数
	count = 0
	#循环遍历剩余天数
	for i in range(k, k+n%7):
	    #如果周六、周日，则休息
	    if i not in {6, 7}:
	        count += 1
	result = 250*(count + 5*week)
	return result