蓝桥杯算法提高：日期计算

问题描述
　　已知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

#include<iostream>
using namespace std;

const int month[2][13]={{0,31,28,31,30,31,30,31,31,30,31,30,31}
						,{0,31,29,31,30,31,30,31,31,30,31,30,31}};

const int week1[7]={5,6,7,1,2,3,4};
const int week2[7]={5,4,3,2,1,7,6};

bool isLeap(int y){
	if((y%4==0&&y%100!=0)||y%400==0){
		return true;
	}
	else
		return false;
}

int main(){
	int y,m,d;
	cin>>y>>m>>d;
	int count=0;
	
	if(y<2011||(y==2011&&m<11)||(y==2011&&m==11&&d<=11)){
		while(y<2011||(y==2011&&m<11)||(y==2011&&m==11&&d<11)){
			d++;
			count++;
			if(d>month[isLeap(y)][m]){
				m+=1;
				d=1;
			}
			if(m>12){
				y+=1;
				m=1;
			}
		}
		cout<<week2[count%7]<<endl;
	}else{
		int a=2011,b=11,c=11;
		while(a<y||(a==y&&b<m)||(a==y&&b==m&&c<d)){
			c++;
			count++;
			if(d>month[isLeap(y)][m]){
				b+=1;
				c=1;
			}
			if(b>12){
				a+=1;
				b=1;
			}
		}
		cout<<week1[count%7]<<endl;
	}
	return 0;
}