题目描述
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;
}