2017"百度之星"程序设计大赛 - 初赛-今夕何夕

今夕何夕

Accepts: 1345
Submissions: 5533
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 32768/32768 K (Java/Others)
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号的特殊判断。暴力往后找满足条件的年份即可。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstring>
using namespace std;
typedef long long LL;
const int INF=0x3f3f3f3f;

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

int calc(int y,int m,int d)
{
    if(m==1||m==2) m+=12,y--;
    return ((d+2*m+3*(m+1)/5+y+y/4-y/100+y/400)%7);
}

int main()
{
    //cout<<judge(2008)<<" "<<judge(2000)<<" "<<judge(1000)<<endl;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int y,m,d;
        scanf("%d-%d-%d",&y,&m,&d);
        //cout<<y<<" "<<m<<" "<<d<<endl;
        int ans=calc(y,m,d);
        //cout<<ans<<endl;
        for(int i=y+1;;i++)
        {
            if((!judge(i))&&m==2&&d==29) continue;
            if(ans==calc(i,m,d)) {printf("%d\n",i);break;}
        }
    }
    return 0;
}