Python练习-蓝桥题库-P498-回文日期

蓝桥题库-P498-回文日期

题目：P498~回文日期

"""
题目描述
2020 年春节期间，有一个特殊的日期引起了大家的注意：2020 年 2 月 2 日。
因为如果将这个日期按 “yyyymmdd” 的格式写成一个 8 位数是 20200202，恰好是一个回文数。我们称这样的日期是回文日期。

有人表示 20200202 是 “千年一遇” 的特殊日子。对此小明很不认同，因为不到 2 年之后就是下一个回文日期：20211202 即 2021 年 12 月 2 日。

也有人表示 20200202 并不仅仅是一个回文日期，还是一个 ABABBABA 型的回文日期。对此小明也不认同，
因为大约 100 年后就能遇到下一个 ABABBABA 型的回文日期：21211212 即 2121 年 12 月 12 日。算不上 “千年一遇”，顶多算 “千年两遇”。

给定一个 8 位数的日期，请你计算该日期之后下一个回文日期和下一个 ABABBABA 型的回文日期各是哪一天。

输入描述
输入包含一个八位整数 N，表示日期。

对于所有评测用例，10000101≤N≤89991231，保证 N是一个合法日期的 8 位数表示。

输出描述
输出两行，每行 1 个八位数。第一行表示下一个回文日期，第二行表示下一个 ABABBABA 型的回文日期。

"""

涉及知识点

"""
Python内置库 datetime相关类和方法
切片
字符串的replace()
循环 + 标记
python语法糖：连续比较运算 
"""

答案代码

import datetime

str_date = input() # 输入

int_year = int(str_date[:4]) # 拆分输入的年月日，并改为int型
int_month = int(str_date[4:6])
int_day = int(str_date[6:])

date1 = datetime.date(int_year, int_month, int_day) # 创建date对象date1

flag = 1 # 循环结束标记
while flag:
  str_date1 = str(date1).replace("-", "") # 将date1转为str，且去除里面的 - ，得到纯数字组成的日期字符串
  if str_date1 == str_date1[::-1]: # 回文日期判定1
    print(str_date1)
    flag = 0
  else: # 不加else的话，date1会在出循环时多加1天，可能导致错过同时满足两种回文格式的日期
    date1 += datetime.timedelta(days=1) # 日期加一天


flag = 1 # 重置循环标记
while flag:
  str_date1 = str(date1).replace("-", "")
  # 回文日期判定2
  if str_date1[0] == str_date1[2] == str_date1[5] == str_date1[7] and \ 
          str_date1[1] == str_date1[3] == str_date1[4] == str_date1[6]:
      print(str_date1)
      flag = 0
  else:
    date1 += datetime.timedelta(days=1)