题目:
"""如果一个日期中年月日的各位数字之和是完全平方数,则称为一个完全日期。例如: 2021年 6月 5日的各位数字之和为 2 + 0 + 2 + 1 + 6 + 5 = 16 而 16是一个完全平方数,它是 4的平方。所以 2021年 6月 5日是一个完全日期。例如: 2021年 6月23日的各位数字之和为 2 + 0 + 2 + 1 + 6 + 2 + 3 = 16 ,是一个完全平方数。所以 2021年 6月 23日也是一个完全日期。请问,从 2001年 1月 1 日到 2021 年 12月 31日中,一共有多少个完全日期?"""
一、暴力枚举
import mathdef digit_sum(year, month, day):# print(type(f"{year}{month:02d}{day:02d}")) # 类型:strreturn sum(map(int, f"{year}{month:02d}{day:02f}"))def is_perfect_date(sum):return int(math.sqrt(sum)) ** 2 == sumdef count_perfect_dates():count = 0for year in range(2001, 2022):for month in range(1, 13): max_day = 31 # 大月if month in [4,6,9,11]: # 小月max_day = 30 elif month == 2:if (year % 4 == 0 and year % 100 !=0) or (year % 400 == 0):max_day = 29else:max_day = 28for day in range(1, max_day + 1):if is_perfect_date(digit_sum(year, month, day)):count += 1return countprint(count_perfect_dates())"""
格式 02d 的解释:
0: 指定如果数字的位数不足两位时,前面会用零填充。
2: 指定输出数字的最小宽度是 2 位。
d: 代表输出的是一个 整数(decimal integer)"""
二、Datetime库
import math
from datetime import date, timedeltadef is_perfact_date(sum):return int(math.sqrt(sum)) ** 2 == sumdef digit_num(cur_date): # 传入日期对象return sum(map(int, f"{cur_date.year}{cur_date.month}{cur_date.day}"))def count_perfect_dates():start_date = date(2001, 1, 1)end_date = date(2021, 12, 31)current_date = start_datecount = 0while current_date <= end_date:if is_perfact_date(digit_num(current_date)):count += 1current_date += timedelta(days=1)return countprint(count_perfect_dates())