回文日期处理方法

题目描述

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\le N\le 89991231$，保证 N 是一个合法日期的 8 位数表示。

20200202

输出

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

20211202
21211212

C++

#include <iostream>
#include <cstring>
using namespace std;
int months[13] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
// 判断回文日期是否合法
bool check_valid1(int date) {
    int year = date / 10000;
    int month = date % 10000 / 100;
    int day = date % 100;
    
    if (month < 1 || month > 12) return false;
    if (day == 0 || month != 2 && day > months[month]) return false;
    if (month == 2) {
        int leap = (year % 100 && year % 4 == 0) || (year % 400); // 闰年判断表达式
        if (day > months[month]+leap) return false;
    }
    return true;
}
// 判断是否是ABABBABA
bool check_valid2(int date) {
	string s = to_string(date);
    if(s[0]!=s[2] || s[1]!= s[3] || s[0] == s[1]) return false;
    return true;
}

int main() {
	int date1;
	cin >> date1;
	int res1 = 0, res2 = 0;
	for (int i = 1000; i < 10000; i++) {
		int date = i, x = i;
		while (x) date = date * 10 + x % 10, x /= 10;
		if (date > date1 && check_valid1(date)) {
			if (!res1)res1 = date;
			if (check_valid2(date) && !res2) res2 = date;
		}	
	}
	cout << res1 << endl << res2 << endl;
	return 0;
}