蓝桥杯 算法提高 日期计算

已知2011年11月11日是星期五，问YYYY年MM月DD日是星期几？注意考虑闰年的情况。尤其是逢百年不闰，逢400年闰的情况。 输入只有一行 YYYY MM DD 输出只有一行 数据规模和约定 1599 = YYYY = 2999 1 = MM = 12 1 = DD = 31，且确保测试样例中YYYY年MM月DD日是一个合理日期 1 = W = 7，分别代表周一到周日 2011 11 11 5

c++：

#include iostream 

#include cstring 

#include cstdio 

using namespace std;

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

bool isRun(int year){//判断是否为闰年

 if(year % 4 == 0 year % 100 != 0 || year % 400 == 0)

 return true;

 return false;

int calDay(int year){

 int index = 0;

 if(!isRun(year)) index = 1;

 int tot = 0;

 for(int i=1; i ++i)

 tot += a[index][i];

 return tot;

int Day(int y, int m, int d){//计算从0年0月0日到y年m月d日的时间天数 

 int ans = 0;

 for(int i=1; i ++i)

 ans+=calDay(i);

 int index = 0;

 if(!isRun(y)) index = 1;

 for(int i=1; i ++i)

 ans += a[index][i];

 ans += d;

 return ans;

int main(){

 int y, m, d;

 cin y m d;

 int diff = Day(y, m, d) - Day(2011, 11, 11);//已知2011-11-11这一天是星期5 

 if(diff 0) cout (4+diff%7)%7+1 endl;

 else{

 for(int i=1; i ++i)

 if((i-1 + -diff%7)%7 + 1 == 5){

 cout i endl;

 break;

 return 0;

}