蓝桥杯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

代码实现

import os
import sys
data = int(input())
temp1=0
temp2 = 0

def fun(data):  #用来判断是否为满足第二个条件，即ABABBABA
  data = str(data)
  if(data[0] == data[2] and data[1] == data[3]):
    return 1
  else :
    return 0

def tiao(nian,yue,ri): #用来判断当前遍历的数字是否符合日期的条件
  if  yue >12 : #若月份大于12，日期大于对应的日期，则进行跳跃
   return 8600 
  if  ri >31: 
    return 70
  if yue == 0 or (ri == 0):
    return 0
  run = (nian%4 == 0 and nian%100 !=0) or nian %400 == 0 #判断是否为闰年
  if  run == 1 and yue == 2:
    if ri>28:
      return 0
  if  run == 0 and yue == 2:
    if ri>29:
      return 0
  if  yue == 4 and ri>30:  #根据不同的月份处理不同的情况
    return 0
  if  yue == 6 and ri>30:
    return 0
  if  yue == 9 and ri>30:
    return 0
  if  yue == 11 and ri>30:
    return 0
  if  ri>21:
    return 0
  return 1

i = data+1
while(i<100000000): #用while遍历，相比于for range，可以进行跳跃，缩短时间
  nian = int(i/10000)
  yue = i%10000
  yue = int(yue/100)
  ri = i%100  #分别得到数字对应的年，月，日
  if tiao(nian,yue,ri) == 8600:
    i+=8700
    i-=i%100
    i+=1
    continue
  elif tiao(nian,yue,ri) == 70:
    i+=70
    i-=i%100
    i+=1
  elif tiao(nian,yue,ri) == 0 :
    i+=1
    continue
  if temp1 == 0 and str(i)[::-1] == str(i): #符合对应条件则输出
    temp1 = i
  if temp2 == 0 and str(i)[::-1] == str(i) and fun(i) :
    temp2 = i
  if temp2 != 0 and temp1 != 0:  #都找到了则直接退出循环进行输出
    break
  i+=1


print('{}\n{}'.format(temp1,temp2))

总结

暴力解法，该题较容易出现超时的现象，所以需要合理剪枝，从而解题。