PTA Python 7-2 今天过后，明天是几号啊

输入今天的日期，输出明天的日期。

提示：每年的1月、3月、5月、7月、8月、10月、12月有31天，4月、6月、9月、11月有30天，闰年的2月有29天，非闰年的2月是28天。闰年的年号：能被4整除但不能被100整除，或者能被400整除。

输入格式:

按照指定的格式输入今天的日期，如：2009-5-12

输出格式:

按照指定的格式输出明天的日期，如：2009-5-13

输入样例1:

2009-5-12

输出样例1:

2009-5-13

输入样例2:

2009-5-31

输出样例2:

2009-6-1

输入样例3:

2009-12-31

输出样例3:

2010-1-1

答案:

# interest=money×(1+rate) year −money
import math

arr = input()
date = arr.split("-")  # 2009-5-12
for x in range(0, 3):
    date[x] = int(date[x])
lest = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]


def month(date):
    if (date[0] % 400 == 0) or (date[0] % 4 == 0 and date[0] % 100 != 0):
        lest[1] = 29
        if date[2] + 1 > lest[date[1] - 1]:
            if date[1] == 12:
                date[1] = 1
                date[0] += 1
                date[2] = 1
            else:
                date[1] += 1
                date[2] = 1
        else:
            date[2] += 1
    else:
        lest[1] = 28
        if date[2] + 1 > lest[date[1] - 1]:
            if date[1] == 12:
                date[1] = 1
                date[0] += 1
                date[2] = 1
            else:
                date[1] += 1
                date[2] = 1
        else:
            date[2] += 1


month(date)
print(f"{date[0]}-{date[1]}-{date[2]}")