世纪末的星期python实现

1.问题

曾有邪教称1999年12月31日是世界末日。当然该谣言已经不攻自破。
还有人称今后的某个世纪末的12月31日，如果是星期一则会....

有趣的是，任何一个世纪末的年份的12月31日都不可能是星期一!! 

于是，“谣言制造商”又修改为星期日......

1999年的12月31日是星期五，请问：未来哪一个离我们最近的一个世纪末年（即xx99年）的12月31日正好是星期天（即星期日）？

请回答该年份（只写这个4位整数，不要写12月31等多余信息）

2.思路及代码

1999.12.31是周五，那距离其天数为7的倍数即永远为周五。
多两天为周日。

day = -2
for i in range(2000,10000):   
    if i%400==0 or (i%4==0 and i%100!=0):
        day += 366
    else:
        day += 365

    if str(i)[-2:] == '99':
        if day%7==0:
            print(i)
        

用库

import datetime


def solve():
    for i in range(1999,10000,100):
        a = datetime.date(i, 12, 31)
        if a.isoweekday() == 7:
            print(i)
            break


solve()