Python——回文日期

题目描述

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 型的回文日期。

例如：

输入
20200202
输出
20211202
21211212

# coding=utf-8
import calendar

# 得到所有回文日期（可能存在不合法的情况）
month_31 = [1,3,5,7,8,10,12]
month_30 = [4,6,9,11]
length = 0
is_print = 2
ideal_data = []

# 翻转函数
def Revert(a):
    s = a[1]+a[0]
    return s

def Connect(a,b):
    for j1 in range(1, b+1):
        if j1 == 10 or j1 == 20 or j1 == 30: continue
        if a < 10 and j1 < 10:
            ideal_data.append( int(Revert('0' + str(j1)) + Revert('0' + str(a)) + '0' + str(a) + '0' + str(j1)) )
        if a < 10 and j1 > 10:
            ideal_data.append( int(Revert(str(j1)) + Revert('0' + str(a)) + '0' + str(a) + str(j1)) )
        if a >= 10 and j1 < 10:
            ideal_data.append( int(Revert('0' + str(j1)) + Revert(str(a)) + str(a) + '0' + str(j1)) )
        if a >= 10 and j1 > 10:
            ideal_data.append( int(Revert(str(j1)) + Revert(str(a)) + str(a) + str(j1)) )

for i in range(1,13):
    if i in month_31:
        Connect(i, 31)
    if i in month_30:
        Connect(i, 30)
    if i == 2:
        Connect(i, 29)
else:
    # 输入合法的八位数日期
    N = int(input())
    ideal_data.sort()
    while length < len(ideal_data):
        length += 1
        current = ideal_data[length-1]
        # 去除不合法数：不是闰年时，二月为29天
        if (not calendar.isleap(ideal_data[length-1]/10000)) and (ideal_data[length-1] % 100 == 29): continue
        if N < ideal_data[length-1]:
            if is_print == 2:
                print(ideal_data[length-1])
                is_print -= 1
            #判断是否为ABABBABA类型 此处取前四位更合适，避免出现后四位为"0101"的情况
            #需要取商的整数部分
            ab = ideal_data[length-1] // 10000
            if ab % 100 == ab // 100:
                print(ideal_data[length-1])
                is_print -= 1
        if is_print == 0: break