蓝桥杯 2020 第7题：回文日期

题目

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

代码

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

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

bool isLeap(int y) {
	return (y % 4 == 0 && y % 100 != 0) || y % 400 == 0;
}

string pin(int y, int m, int d) {
	string s, s1;
	s1 = to_string(y);
	s += s1;
	s1 = to_string(m);
	if (s1.size() == 1) {
		s1 = "0" + s1;
	}
	s += s1;
	s1 = to_string(d);
	if (s1.size() == 1) {
		s1 = "0" + s1;
	}
	s += s1;
	return s;
}

bool huiwen(int y,int m,int d) {
	string s = pin(y, m, d);
	for (int i = 0; i < 4; i++) {
		if (s[i] != s[7 - i])
			return false;
	}
	return true;
}

bool AB(int y, int m, int d) {
	string s = pin(y, m, d);
	int A = s[0];
	int B = s[1];
	if (s[2] == A && s[3] == B && s[4] == B && s[5] == A && s[6] == B && s[7] == A) {
		return true;
	}
	return false;
}

int main() {
	int n;
	cin >> n;
	int y = n / 10000, m = n % 10000 / 100, d = n % 100;
	int f1 = 0, f2 = 0;
	while (true) {	
		d++;
		if (isLeap(y))
			month[2] = 29;
		else
			month[2] = 28;
		if (d > month[m]) {
			m++;
			d = 1;
		}
		if (m > 12) {
			y++;
			m = 1;
		}
		if (f1 == 0) {
			if (huiwen(y, m, d)) {
				string s = pin(y, m, d);
				int time = stoi(s);
				cout << time << endl;
				f1 = 1;
			}
		}
		if (f2 == 0) {
			if (AB(y, m, d)) {
				string s = pin(y, m, d);
				int time = stoi(s);
				cout << time << endl;
				f2 = 1;
			}
		}
		if (f1 && f2) {
			break;
		}
	}

	return 0;
}