回文日期 -蓝桥杯-暴力

题目描述

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

输出描述

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

枚举

#include <bits/stdc++.h>

using namespace std;

unordered_set<int> s1 = {1, 3, 5, 7, 8, 10, 12};
unordered_set<int> s2 = {4, 6, 9, 11};

// 判断是否是闰年
bool isLeap(int year) {
	return (year % 4 == 0 && year % 100 != 0) || (year % 400 == 0);
}

bool check(int year, int month, int day) {
	if (month > 12 || month == 0 || day == 0 || day > 31) {
		return false;
	}
	if (s1.count(month)) {
		return day <= 31;
	}
	if (s2.count(month)) {
		return day <= 30;
	}
	if (month == 2) {
		if (isLeap(year)) {
			return day <= 29;
		}
		return day <= 28;
	}
	return false;
}

int main() {
	int n;
	cin >> n;
	bool flag1 = false, flag2 = false;
	for (int i = n + 1; i <= 99999999; i ++) {
		int year = i / 10000;
		int month = (i % 10000) / 100;
		int day = i % 100;
		int a = i % 10;
		int b = (i / 10) % 10;
		int c = (i / 100) % 10;
		int d = (i / 1000) % 10;
		int e = (i / 10000) % 10;
		int f = (i / 100000) % 10;
		int g = (i / 1000000) % 10;
		int h = (i / 10000000) % 10;

		if (h == a && g == b && f == c && e == d && !flag1) {
			if (check(year, month, day)) {
				cout << i << endl;
				flag1 = true;
			}
		}

		if (h == f && f == c && c == a && g == e && e == d && d == b && !flag2) {
			if (check(year, month, day)) {
				cout << i << endl;
				flag2 = true;
			}
		}
		if (flag1 && flag2) {
			break;
		}
	}

	return 0;
}