HDU 6112 今夕何夕

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6112

今夕何夕

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 1696 Accepted Submission(s): 603

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

Source
2017”百度之星”程序设计大赛 - 初赛（A）

做这种日期题，不要漏了闰年2月29这种特殊情况一般就没有什么问题
刚开始一直没有想到2月29这个特殊测试数据。

正确代码：

#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;
typedef long long LL;
int month[15]={0,31,28,31,30,31,30,31,31,30,31,30,31};
int judge(int n)
{//判断是否为闰年
    if((n%4==0&&n%100!=0)||n%400==0)
        return 1;
    return 0;
}
int js(int yy,int mm,int dd)
{//计算所给日期到年末的天数
    int ans=0;
    if(judge(yy)==1&&mm==2)
        ans+=29-dd;
    else
        ans+=month[mm]-dd;
    for(int i=mm+1;i<=12;i++)
    {
        if(judge(yy)==1&&i==2)
            ans+=29;
        else
            ans+=month[i];
    }
    return ans;
}
int js1(int yy,int mm,int dd)
{//计算年初到所给日期的天数
    int ans=0;
    for(int i=1;i<mm;i++)
    {
        if(judge(yy)==1&&i==2)
            ans+=29;
        else
            ans+=month[i];
    }
    ans+=dd;
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    getchar();
    while(t--)
    {
        int sum=0,x=0,y=0,z=0,i;
        char a[20];
        scanf("%s",a);
        x=(a[0]-'0')*1000+(a[1]-'0')*100+(a[2]-'0')*10+a[3]-'0';
        y=(a[5]-'0')*10+a[6]-'0';
        z=(a[8]-'0')*10+a[9]-'0';
        int ans=0;
        ans+=js(x,y,z);
        //要分为两种情况
        //所给日期是闰年的2月29，那么下一个2月29就是下一个闰年才会有（不一定隔四年）
        //所给日期不是闰年的2月29，那就一年一年的算了
        if(y==2&&z==29)
        {
            for(i=x+4;;i+=4)
            {
                ans+=365*3;
                int sum=js1(i,y,z);
                if(judge(i)==1)
                {
                    if((sum+ans)%7==0)
                    {
                        printf("%d\n",i);
                        break;
                    }
                    else
                        ans+=366;
                }
                else
                    ans+=365;
            }
        }
        else
        {
            for(i=x+1;;i++)
            {
                int sum=js1(i,y,z);
                if((sum+ans)%7==0)
                {
                    printf("%d\n",i);
                    break;
                }
                else
                {
                    if(judge(i)==1)
                        ans+=366;
                    else
                        ans+=365;
                }
            }
        }
    }
    return 0;
}

第二种方法：（只改动了一个小部分）

#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;
typedef long long LL;
int month[15]={0,31,28,31,30,31,30,31,31,30,31,30,31};
int judge(int n)
{
    if((n%4==0&&n%100!=0)||n%400==0)
        return 1;
    return 0;
}
int js(int yy,int mm,int dd)
{
    int ans=0;
    if(judge(yy)==1&&mm==2)
        ans+=29-dd;
    else
        ans+=month[mm]-dd;
    for(int i=mm+1;i<=12;i++)
    {
        if(judge(yy)==1&&i==2)
            ans+=29;
        else
            ans+=month[i];
    }
    return ans;
}
int js1(int yy,int mm,int dd)
{
    int ans=0;
    for(int i=1;i<mm;i++)
    {
        if(judge(yy)==1&&i==2)
            ans+=29;
        else
            ans+=month[i];
    }
    ans+=dd;
    return ans;
}
int calcu(int yy,int y)
{
    int ans=0;
    for(int i=yy;i<y;i++)
    {
        if(judge(i))ans+=366;
        else ans+=365;
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    getchar();
    while(t--)
    {
        int x,y,z,i;
        scanf("%d-%d-%d",&x,&y,&z);
        int ans=0;
        if(y==2&&z==29)
        {
            int pre=x;
            for(int i=x+4;;i+=4)
            {
                if(judge(i)==0)continue;
                ans+=calcu(pre,i);
                if(ans%7==0)
                {
                    printf("%d\n",i);
                    break;
                }
                pre=i;
            }
        }
        else
        {
            ans+=js(x,y,z);
            for(i=x+1;;i++)
            {
                int sum=js1(i,y,z);
                if((sum+ans)%7==0)
                {
                    printf("%d\n",i);
                    break;
                }
                else
                {
                    if(judge(i)==1)
                        ans+=366;
                    else
                        ans+=365;
                }
            }
        }
    }
    return 0;
}