HDU 6112 今夕何夕（模拟）

Problem Description 今天是2017年8月6日，农历闰六月十五。 小度独自凭栏，望着一轮圆月，发出了“今夕何夕，见此良人”的寂寞感慨。 为了排遣郁结，它决定思考一个数学问题：接下来最近的哪一年里的同一个日子，和今天的星期数一样？比如今天是8月6日，星期日。下一个也是星期日的8月6日发生在2023年。 小贴士：在公历中，能被4整除但不能被100整除，或能被400整除的年份即为闰年。 Input 第一行为T，表示输入数据组数。 每组数据包含一个日期，格式为YYYY-MM-DD。 1 ≤ T ≤ 10000 YYYY ≥ 2017 日期一定是个合法的日期 Output 对每组数据输出答案年份，题目保证答案不会超过四位数。 Sample Input 3 2017-08-06 2017-08-07 2018-01-01 Sample Output 2023 2023 2024

 

分三种情况考虑，第一种月份小于等于2并且天数小于29，第二种月份大于2，第三种日期为2月29的。

#include <iostream>
#include <cstdio>
using namespace std ;
int year, month, day, n;
bool judge_year(int y) {
    if(y%400==0 || (y%4==0&&y%100!=0)) {
        return true;
    }
    return false;
}
int main() {
    scanf("%d", &n);
    char a, b;
    while(n--) {
        scanf("%d%c%d%c%d", &year, &a, &month, &b, &day);
        int ans = 0;
        if(month < 2||(month == 2 && day < 29)) {
	        for(int i = year+1; i <= 9999; i++) {
	            if(judge_year(i - 1))
                                ans++;
	            ans++;
	            if(ans%7==0 && ans>0) {
                    printf("%d\n",i );
                    break;
	            }
	        }
        }
        if(month > 2) {
			for(int i = year+1; i <= 9999; i++) {
			    if(judge_year(i))
                                ans++;
				ans++;
				if(ans%7==0 && ans>0) {
                    printf("%d\n",i);
                    break;
				}
			}
        }
        if(month==2 && day==29) {
            for(int i = year+1; i <= 9999; i++) {
				if(judge_year(i))
                                ans++;
				ans++;
				if(ans%7==0 && ans>0 && judge_year(i)) {
                    printf("%d\n",i);
                    break;
				}
			}
        }
    }
    return 0;
}