求n天后的日期

问题描述 :

写一个函数，传入年月日，计算它的第二天，并返回该日期。由用户输入年月日和一个n值，使用前述函数，计算该日期加n天的日期为多少。

输入说明 :

输入year,month,day和n共4个正整数，以空格分隔。n的值不超过2000。

输出说明 :

输出计算得到的结果年月日共3个正整数，整数之间以一个空格分隔，行首与行尾无多余空格。

#include <cstdio>
#include <cmath>
using namespace std;
int leap(int y)
{
	if (y % 4 == 0 && y % 100 != 0 || y % 400 == 0)
	{
		return 1;
	}
	return 0;
}
int main(){
	int year,month,day,n;
	int mon[13]={0,31,28,31,30,31,30,31,31,30,31,30,31};
	scanf("%d%d%d%d",&year,&month,&day,&n);
		if(leap(year))
		{
            mon[2]=29;
		}
		else
			mon[2]=28;


		for(int i=1;i<=n;i++){
			day+=1;
			if(day>mon[month]){
				day=day-mon[month];
				month++;
				if(month>12){
					year++;
					month=1;
					if(leap(year)){
						mon[2]=29;
					}
					else
						mon[2]=28;
				}
			}
		}
		printf("%d %d %d",year,month,day);

	return 0;
}