蓝桥01 跑步锻炼 —— datatime

跑步锻炼

题目描述
本题为填空题，只需要算出结果后，在代码中使用输出语句将所填结果输出即可。

小蓝每天都锻炼身体。

正常情况下，小蓝每天跑 11 千米。如果某天是周一或者月初（11 日），为了激励自己，小蓝要跑 22 千米。如果同时是周一或月初，小蓝也是跑 22 千米。

小蓝跑步已经坚持了很长时间，从 20002000 年 11 月 11 日周六（含）到 20202020 年 1010 月 11 日周四（含）。请问这段时间小蓝总共跑步多少千米？

运行限制
最大运行时间：1s
最大运行内存: 128M

import datetime

start = datetime.date(2000, 1, 1)
end = datetime.date(2020, 10, 1)
days = datetime.timedelta(days=1)
ans = 0

while end >= start:
    if start.day == 1 or start.weekday() == 0:
        ans += 2
    else:
        ans += 1
    start += days
print(ans)

鄙人丑陋的代码：

import os
import sys

# 请在此输入您的代码

def check(date):
  year = date[0]
  month = date[1]
  day = date[2]
  week = date[3]
  if week == 1 or day ==1:
    return True
  else:
    return False

def nextday(date):
  year = date[0]
  month = date[1]
  day = date[2]
  week = date[3]
  nextdate = []
  if month == 12 and day == 31: # 如果是每年的最后一天
      year = year+1
      month = 1
      day = 1
  else: # 如果不是每年的最后一天
    if year%400 == 0 or (year%4==0 and year%100!=0):
      monthday = [31,29,31,30,31,30,31,31,30,31,30,31]
      if day == monthday[month-1]: # 如果是闰年每月（非12月）的最后一天
        month = month+1
        day = 1
      else: # 如果不是闰年每月（非12月）的最后一天
        day = day + 1
    else:
      monthday = [31,28,31,30,31,30,31,31,30,31,30,31]
      if day == monthday[month-1]: # 如果是每月（非12月）的最后一天
        month = month+1
        day = 1
      else: # 如果不是每月（非12月）的最后一天
        day = day + 1
  nextdate.append(year)
  nextdate.append(month)
  nextdate.append(day)
  nextdate.append(week%7+1)
  return nextdate

number = 0
date = [2000,1,1,6]
while(True):
  if check(date)==True:
    number+=2
  else:
    number+=1
  date = nextday(date)
  if date == [2020,10,2,5]:
    break
print(number)