hatter/python-tests
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: update leap_year.py

Browse Source
This commit is contained in:
Hatter Jiang
2025-09-05 00:49:30 +08:00
parent 21400b1d01
commit 239a0b50d0
1 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
leap_year.py Normal file → Executable file
View File

@@ -4,7 +4,38 @@ def is_leap_year(year: int):
return year % 4 == 0 and year % 100 != 0 or year % 400 == 0
def get_day_of_month(year, month: int) -> int:
match month:
case 1 | 3 | 5 | 7 | 8 | 10 | 12:
return 31
case 4 | 6 | 9 | 11:
return 30
case 2 if is_leap_year(year):
return 29
case 2:
return 28
case _:
raise Exception(f"invalid month: {month}")
def get_day_of_month2(year, month: int) -> int:
if [1, 3, 5, 7, 8, 10, 12].__contains__(month):
return 31
elif [4, 6, 9, 11].__contains__(month):
return 30
elif month == 2 and is_leap_year(year):
return 29
elif month == 2:
return 28
else:
raise Exception(f"invalid month: {month}")
if __name__ == "__main__":
print(1582, is_leap_year(1582))
print(2000, is_leap_year(2000))
print(2100, is_leap_year(2100))
print(2000, 1, '->', get_day_of_month(2000, 1), get_day_of_month2(2000, 1))
print(2000, 2, '->', get_day_of_month(2000, 2), get_day_of_month2(2000, 2))
print(2001, 2, '->', get_day_of_month(2001, 2), get_day_of_month2(2001, 2))