《算法笔记》3.4小节——入门模拟-＞日期处理 问题 E: 日期累加

问题 E: 日期累加

时间限制 : 1.000 sec 内存限制 : 32 MB

题目描述

设计一个程序能计算一个日期加上若干天后是什么日期。

输入

输入第一行表示样例个数m，接下来m行每行四个整数分别表示年月日和累加的天数。

输出

输出m行，每行按yyyy-mm-dd的个数输出。

样例输入

1
2008 2 3 100

样例输出

2008-05-13

#include <iostream>

using namespace std;
int a[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}
};

int isLeap(int year) {
    if (year % 400 == 0 || year % 4 == 0 && year % 100 != 0)
        return 1;
    else
        return 0;
}

int main() {

    int n;
    while (cin >> n) {
        while (n--) {
            int sum = 0;
            int year, total_day, month, day;
            cin >> year >> month >> day >> total_day;
            int leap = 0;
            leap = isLeap(year);
            while (sum != total_day) {
                sum++;
                day++;
                if (day == a[leap][month] + 1) {
                    month++;
                    day = 1;
                }
                if (month == 13) {
                    month = 1;
                    year++;
                    leap = isLeap(year);
                }
            }
            printf("%04d-%02d-%02d\n", year, month, day);
        }
    }
    return 0;
}