【蓝桥真题】 回文日期

题目描述
2020 年春节期间，有一个特殊的日期引起了大家的注意：2020 年 2 月 2 日。

因为如果将这个日期按 yyyymmdd 的格式写成一个 8 位数是 20200202，恰好是一个回文数。

我们称这样的日期是回文日期。

有人表示 20200202 是“千年一遇” 的特殊日子。

对此小明很不认同，因为不到 2 年之后就是下一个回文日期：20211202 即 2021 年 12 月 2 日。

也有人表示 20200202 并不仅仅是一个回文日期，还是一个 ABABBABA 型的回文日期。

对此小明也不认同，因为大约 100 年后就能遇到下一个 ABABBABA 型的回文日期：21211212 即 2121 年 12 月 12 日。

算不上“千年一遇”，顶多算“千年两遇”。

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

注意：下一个回文日期和下一个 ABABBABA 型的回文日期可能是同一天。

ABABBABA 型的回文日期，需要满足 A≠B。

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

输出格式
第一行表示下一个回文日期，
第二行表示下一个 ABABBABA 型的回文日期。

输入样例
20200202

输出样例
20211202
21211212

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

解题思路
枚举年份a，将其反转得到b，a+b得到一个回文数
用substr分离出月份和天数，在判断月份和天数是否合法；

代码

#include <iostream>
#include <algorithm>
using namespace std;

int days[] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};

bool check(int year)
{
	return year % 400 == 0 || year % 4 == 0 && year % 100 != 0; //闰年2月多一天
}

int get_day(int year, int month)
{
	if(month == 2) return 28 + check(year);
	return days[month];
}

int main()
{
	int n;
	cin >> n;
	
	string ans1, ans2;  //ans1回文数 ans2为ababbaba形回文数
	bool flag1 = false, flag2 = false; //两个哨兵
	for (int i = n / 10000; i <= 9999; i ++) //枚举年份
	{
		string a = to_string(i); //to_string整转字
		string b = a; 
		reverse(b.begin(), b.end()); //反转b
		if(a + b <= to_string(n)) continue; //得到日期需大于输入日期
		
		int month = stoi(b.substr(0, 2)); //stoi字转整 substr分理出月份和天
		int day = stoi(b.substr(2, 2));
		if(month < 1 || month > 12) continue; //判断是否合法
		if(day < 1 || day > get_day(i, month)) continue;
		
		string s1 = a.substr(0, 2); //作为判断ababbaba型
		string s2 = a.substr(2, 2);
		if(!flag1) ans1 = a + b, flag1 = true;
		if(!flag2 && s1 == s2 && s1[0] != s1[1]) ans2 = a + b, flag2 = true; //判断ababbaba型

		if(flag1 && flag2) break; //得到两个回文数结束循环
	}
	
	cout << ans1 << endl;
	cout << ans2 << endl;
	return 0;
}