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

32017-08-062017-08-072018-01-01

 

 

Sample Output

202320232024

 

调了半天bug，结果是认为

2月29的时候一定是4*7=28年后，实际上并非如此

 

代码

#include<bits/stdc++.h>
#include<algorithm>
#include<cstring>
#define ms(a) memset(a,0,sizeof(a))
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;

int jude(int y)
{
    if(y%400==0||(y%100&&y%4==0))
        return 1;
    return 0;
}

int main()
{
    int t;
    scanf("%d",&t);

    while(t--)
    {
        int year,moth,day;
        scanf("%d-%d-%d",&year,&moth,&day);

        int c = 0;
        if((moth<=2))
        {
            for(int i=year;; i++)
            {
                if(jude(i))
                    c+=2;
                else
                    c++;
                if(c%7==0)
                {
                    if( moth==2&&day==29 && jude(i+1))
                    {
                        printf("%d\n",i+1);
                        break;
                    }
                    else if(!(moth==2&&day==29))
                    {
                        printf("%d\n",i+1);
                        break;
                    }
                }
            }
        }
        else
        {
            for(int i=year+1;; i++)
            {
                if(jude(i))
                    c+=2;
                else
                    c++;
                if(c%7==0)
                {
                    printf("%d\n",i);
                    break;
                }
            }
        }
    }

    return 0;
}