Redraiment的遭遇——Python

Lee 的老家住在工业区，日耗电量非常大。今年 7 月，传来了不幸的消息，政府要在 7、8 月对该区进行拉闸限电。政府决定从 7月 1 日起停电，然后隔一天到 7 月 3 日再停电，再隔两天到 7 月 6 日停电，依次下去，每次都比上一次长一天。Lee 想知道自己到家后到底要经历多少天倒霉的停电。请编写程序帮他算一算。

任务：实现停电停多久问题关键算法。

注意：从键盘输入放假日期、开学日期，日期限定在 7、8 月份，且开学日期大于放假日期，然后在屏幕上输出停电天数。

提示：可以用数组标记停电的日期。

def td():
    sm = int(input('请输入起始月份：'))
    sd = int(input('请输入起始天数：'))
    em = int(input('请输入结束月份：'))
    ed = int(input('请输入结束天数：'))
    b = [i for i in range(1, 63)]
    if sm == 7:
        b = b[sd - 1:]
    else:
        b = b[sd - 1 + 31:]
    if em == 7:
        b = b[0:ed - sd + 1]
    else:
        b = b[0:31 - sd + 1 + ed]
    a, c, res = 1, 1, 0
    while a <= 62:
        a = int((c + 1) * c / 2)
        c += 1
        if a in b:
            res += 1
            print(a, end=',')
    print(f'
天数：{res}')