算法提高 日期计算

http://lx.lanqiao.org/problem.page?gpid=T265

算法提高 日期计算  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　已知2011年11月11日是星期五，问YYYY年MM月DD日是星期几？注意考虑闰年的情况。尤其是逢百年不闰，逢400年闰的情况。

输入格式

　　输入只有一行
　　YYYY MM DD

输出格式

　　输出只有一行
　　W

数据规模和约定

　　1599 <= YYYY <= 2999
　　1 <= MM <= 12
　　1 <= DD <= 31，且确保测试样例中YYYY年MM月DD日是一个合理日期
　　1 <= W <= 7，分别代表周一到周日

样例输入

2011 11 11

样例输出

5

 

分析：

以某一个点为参照，向前或者向后计算，注意闰年2月的计算。

 

AC代码：

 

#include<iostream>
#include <stdio.h>
#include<cmath>
using namespace std;

bool leap(int year)
{
    if(year%4==0&&year%100!=0)
    return true;
    if(year%400==0)
    return true;
    return false;
} 

int getYD(int year)
{
    int sum=0;
    int i;
    if(year<=2011)
    {
        for(i=year;i<2011;i++)
        {
            if(leap(i))
            sum+=366;
            else
            sum+=365;    
        }
    }
    if(year>2011)
    {
        for(i=2011;i<year;i++)
        {
            if(leap(i))
            sum+=366;
            else
            sum+=365;
        }
    }
    return sum;
}

int getMD(int year,int mon,int day)
{
    int i;
    int sum=0;
    for(i=1;i<mon;i++)
    {
        if(i==1||i==3||i==5||i==7||i==8||i==10||i==12)
        sum+=31;
        if(i==4||i==6||i==9||i==11)
        sum+=30;
        if(i==2&&leap(year))
        sum+=29;
        if(i==2&&!leap(year))
        sum+=28;
    }
    sum=sum+day-1;
    return sum;
}



int main()
{
    //freopen("1.in ", "r" , stdin);
    //freopen("1.out" , "w" , stdout);

    int year,mon,day,sum;
    while(cin>>year>>mon>>day)
    {
        sum=getYD(year);
        //cout<<getMD(year,mon,day)<<endl;;
        if(year<2011)
        {
            sum=sum-getMD(year,mon,day);
            //cout<<sum<<endl;
            cout<<(sum%7==6?7:abs(6-(sum%7)))<<endl;
        }
        else
        {
            sum=sum+getMD(year,mon,day);
            //cout<<sum<<endl;
            cout<<((6+(sum%7))>7?((6+(sum%7))%7):(6+(sum%7)))<<endl;
        }
        
        
    }
    return 0;
}