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;
}