算法提高 日期计算

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

问题描述

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

 

AC  code：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define LL long long
using namespace std;
int mon[14]={0,31,28,31,30,31,30,31,31,30,31,30,31};
int cal(int y,int m,int d)
{
	int cnt=d;
	for(int i=1;i<m;i++)
	{
		cnt+=mon[i];
	}
	if(y%400==0||(y%100!=0&&y%4==0))
	{
		if(m>2)
		{
			cnt++;
		}
	
	}
	return cnt;
}
bool jude(int y)
{
	return (y%400==0||(y%100!=0&&y%4==0));
}
int y,m,d,ans,fg;
int main()
{
	while(cin>>y>>m>>d)
	{
		int cnt=cal(y,m,d);
		if(y==2012)
		{
			cnt--;
			if(cnt%7==0)
				ans=7;
			else
			{
				ans=cnt%7;
			}
		}
		else if(y<2012)
		{
			int ty=y;
			int sd=0;
			if(jude(y))
			{
				sd+=366-cnt;	
			}
			else
			{
				sd+=365-cnt;
			}
			ty++;
			while(ty<2012)
			{
				if(jude(ty))
				{
					sd+=366;
				}
				else
				{
					sd+=365;
				}
				ty++;
			 }
			 sd++;
			 if(sd%7==0)
			 {
			 	ans=7;
			 }
			 else
			 {
			 	ans=7-sd%7;
			 }	   
		}
		else
		{
			int ty=2012;
			int sd=cnt;
			while(ty<y)
			{
				if(jude(ty))
				{
					sd+=366;
				}
				else
				{
					sd+=365;
				}
				ty++;
			}
			sd--;
			if(sd%7==0)
			{
				ans=7;
			}
			else
			{
				ans=sd%7;
			}	
		}
		cout<<ans<<endl;
		
	}
	return 0;
 }