蓝桥杯 算法提高 日期计算

问题描述

　　已知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

思路：为了计算方便起见，选择一个在1月1日是星期一的年份作为分界线：2018年1月1日。

将输入的日期分为两种情况：1）大于等于2018年的日期；2）小于2018年的日期。

接下来就是计算输入日期距离2018、1、1的天数。

在2018年之前的日期计算出总天数之后求该天是星期几有规律：7-（sum%7）+1;

在2018年之后的日期计算出总天数之后求改天是星期几有规律：sum%7；

#include<cstdio>
#include<cstring>
using namespace std;
int year(int y)
{
    if(y%400==0||y%4==0&&y%100!=0)
        return 1;
    else
        return 0;
}
int month(int m,int flag)
{
    if(m==1||m==3||m==5||m==7||m==8||m==10|m==12)
        return 31;
    else if(m==2&&flag==1)
        return 29;
    else if(m==2&&flag==0)
        return 28;
    else if(m==4||m==6||m==9||m==11)
        return 30;

}
int main()
{
    int y,m,d;
    int flag=0,sum=0,x=0;
    scanf("%d %d %d",&y,&m,&d);
    if(y>=1599&&y<=2999&&m>=1&&m<=12&&d>=1&&d<=31)
    {
        if(y<2018)
        {
            flag=year(y);
            for(int i=2017;i>y;i--)
            {
                if(year(i))
                    sum+=366;
                else
                    sum+=365;
            }
            for(int i=12;i>=m;i--)
                sum+=month(i,flag);
            sum=sum-d+1;
            x=7-(sum%7)+1;
        }
        else if(y>=2018)
        {
            flag=year(y);
            for(int i=2018;i<y;i++)
            {
                if(year(i))
                    sum+=366;
                else
                    sum+=365;
            }
            for(int i=1;i<m;i++)
                sum+=month(i,flag);
            sum+=d;
            x=sum%7;
        }
        printf("%d\n",x);
    }
    return 0;
}