【蓝桥杯 每天一道算法题】回文日期

题目描述

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

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

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

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

输入描述

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

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

输出描述

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

输入输出样例

示例

输入

20200202

输出

20211202
21211212

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

答案

# https://www.lanqiao.cn/problems/498/learning/
import datetime

date = input()
y, m, d = int(date[:4]), int(date[4:6]), int(date[6:])
dd = datetime.date(y, m, d)
flag1 = flag2 = False
ans = [0, 1]
for i in range(9999999):
    # 注意点1：datatime 对时间的操作
    dd = dd + datetime.timedelta(days=1)
    strdd = str(dd).replace('-', '')
    if flag1 and flag2:
        break
    else:
        if not flag1 and strdd == strdd[::-1]:
            ans[0] = strdd
            flag1 = True
        if not flag2 and strdd[0] == strdd[2] == strdd[-1] == strdd[-3] and strdd[1] == strdd[3] == strdd[-2] == strdd[-4]:
            ans[1] = strdd
            flag2 = True
print(ans[0])
print(ans[1])