优化这段代码year, month, day = map(int, input().split()) # 输入年月日，使用 map 函数将其转换为整数if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0: # 判断是否为闰年 days_of_month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31] # 闰年每月天数else: days_of_month = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31] # 平年每月天数if day < days_of_month[month-1]: # 当天不是本月最后一天 day += 1else: # 当天为本月最后一天 day = 1 if month < 12: # 当月不是本年最后一个月 month += 1 else: # 当月为本年最后一个月 month = 1 year += 1print('{:04d} {:02d} {:02d}'.format(year, month, day)) # 输出结果，使用 format 函数将输出格式化为 YYYY MM DD 的形式，不足两位的用 0 填充

可以考虑使用 datetime 模块来简化代码：

from datetime import datetime, timedelta

year, month, day = map(int, input().split())
date = datetime(year=year, month=month, day=day)
next_date = date + timedelta(days=1)
print(next_date.strftime('%Y %m %d'))

这里使用 datetime 对象来表示日期，通过 timedelta 对象加上一天来计算下一天的日期。最后使用 strftime 方法将日期格式化为指定的字符串形式。