回文日期--蓝桥杯

在线评测地址：https://www.lanqiao.cn/problems/498/learning/

题目描述：

2020 年春节期间，有一个特殊的日期引起了大家的注意：2020 年 2 月 2 日。因为如果将这个日期按 “yyyymmdd” 的格式写成一个 8 位数是 20200202，恰好是一个回文数。我们称这样的日期是回文日期。
有人表示 20200202 是 “千年一遇” 的特殊日子。对此小明很不认同，因为不到 2 年之后就是下一个回文日期：20211202 即 2021 年 12 月 2 日。
也有人表示 20200202 并不仅仅是一个回文日期，还是一个 ABABBABA 型的回文日期。对此小明也不认同，因为大约 100 年后就能遇到下一个 ABABBABA 型的回文日期：21211212 即 2121 年 12 月 12 日。算不上 “千年一遇”，顶多算 “千年两遇”。
给定一个 8 位数的日期，请你计算该日期之后下一个回文日期和下一个 ABABBABA 型的回文日期各是哪一天。
输入描述
输入包含一个八位整数 NNN，表示日期。
对于所有评测用例，10000101≤N≤8999123110000101 \leq N \leq 8999123110000101≤N≤89991231，保证 NNN 是一个合法日期的 8 位数表示。
输出描述
输出两行，每行 1 个八位数。第一行表示下一个回文日期，第二行表示下一个 ABABBABA 型的回文日期。
输入输出样例
示例
输入
20200202
输出
20211202
21211212
运行限制
最大运行时间：1s
最大运行内存: 256M

思路：一个个判断一个8位数是不是回文日期太过麻烦，不如我们直接构造回文日期。我们只要枚举前4位数字1000~8999然后直接构造回文即可得到所有要求范围内的回文串。
例如：得到1001 就可以构造出 10011001这个回文串。
------同时我们也会发现构造出来的回文串不一定满足日期的格式，所以每当我们得到一个回文日期后可以判断一下后四位：月和日是否满足日期格式，如果是就放入数组中。
------最后用查找第一个比输入日期大的回文日期和满足ababbaba型的回文日期输出即可。

#include<iostream>
#include<string>
#include<vector>
using namespace std;
const int s = 1000, e = 8999;
int n;
vector<int> arr;
int month[13] = { 0, 31,28,31,30,31,30,31,31,30,31,30,31 };
bool check(int yy, int mm, int dd) {//是否满足日期格式
	if (mm >= 1 && mm <= 12) {
		if (yy % 400 == 0 || (yy % 4 == 0 && yy % 100 != 0)) 
			month[2] = 29;
			
		if (month[mm] >= dd) return true;
		else return false;
	}
	return false;
}
void init() {//预处理回文日期
	for (int i = s; i <= e; i++) {
		string s = to_string(i);//转string

		for (int j = 3; j >= 0; j--) s += s[j];

		int mm = (s[4] - '0')*10 + (s[5] - '0');//月
		int dd = (s[6] - '0')*10 + (s[7] - '0');//日

		if (check(i, mm, dd)) {
			int k = stoi(s);//转int
			arr.push_back(k);//回文的日期统一存起来
		}
		month[2] = 28;
		
	}//for
}
int upper_bound(int key) {//找到第一个比输入大的回文日期
	int l, r, mid;
	l = 0, r = arr.size();
	while (l <= r) {
		mid = (l + r) >> 1;

		if (arr[mid] > key) r = mid - 1;
		else l = mid + 1;
	}
	return l;
}
int ababbaba(int index) {//寻找满足 ababbaba 格式的回文日期
	for (int i = index; i < arr.size(); i++) {
		string s = to_string(arr[i]);

		if (s[0] == s[2] && s[1] == s[3]) return i;
	}
	return 0;
}
int main() {
	int n;
	cin >> n;
	init();

	int index = upper_bound(n);
	cout << arr[index] << '\n';

	index = ababbaba(index);
	cout << arr[index];
	return 0;
}