2017百度之星资格赛（A)1005 今夕何夕(日期计算)

// 今夕何夕.cpp 运行/限制：15ms/1000ms
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int month[2][13] = { {0,31,28,31,30,31,30,31,31,30,31,30,31},{ 0,31,29,31,30,31,30,31,31,30,31,30,31 } };
int judge(int y) {//判断瑞年
if ((y % 4 == 0 && y % 100 != 0) || y % 400 == 0) {
return 1;
}
return 0;
}
int calPre(int flag, int m, int d) {//flag用于说明瑞年平年，m-d为日期，计算本年该日期前的天数（包括本天）
int re = 0;
for (int i = 1; i < m; i++) {
re += month[flag][i];
}
re += d;
return re;
}
int calBack(int flag, int pre) {//计算本年该日期后的天数（不包括本天）
if (flag) {
return 366 - pre;
}
else {
return 365 - pre;
}
}
int main(){
int t;
int y, m, d;
scanf("%d", &t);
while (t--) {
scanf("%d-%d-%d", &y, &m, &d);
int flag = judge(y);//标记给出日期的瑞年平年
int pre = calPre(flag, m, d);
int back = calBack(flag, pre);
int re = back, i;
if (m == 1) {//一月份
for (i = y + 1;; i++) {
re += pre;
if (re % 7 == 0) {
break;
}
int f = judge(i);
re += f ? 366 - pre : 365 - pre;
}
}
else if(m > 2){//二月后
for (i = y + 1;; i++) {
int f = judge(i);
int p = flag > f ? pre - 1 : (f > flag?pre + 1 : pre);
re += p;
if (re % 7 == 0) {
break;
}
re += f ? 366 - p : 365 - p;
}
}
//二月；二月是个特殊月份，所以对月份分类讨论；
//二月份又得分类讨论日子是不是29号，因为二月由于年数不同可能天数不同，存在开始时间为29号，当前年没有29号的情况
else if(d == 29){//日期为02-29
for (i = y + 1;; i++) {
int f = judge(i);
if (!f) {//该年没29号
re += 365;
continue;
}
re += pre;
if (re % 7 == 0) {
break;
}
re += 366 - pre;
}
}
else {
for (i = y + 1;; i++) {
re += pre;
if (re % 7 == 0) {
break;
}
int f = judge(i);
re += f?366 - pre:365 - pre;
}
}
printf("%04d\n", i);
}
return 0;
}