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

问题 D: 日期类

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

题目描述

编写一个日期类，要求按xxxx-xx-xx 的格式输出日期，实现加一天的操作。

输入

输入第一行表示测试用例的个数m，接下来m行每行有3个用空格隔开的整数，分别表示年月日。测试数据不会有闰年。

输出

输出m行。按xxxx-xx-xx的格式输出，表示输入日期的后一天的日期。

样例输入

2
1999 10 20
2001 1 31

样例输出

1999-10-21
2001-02-01

提示

注意个位数日期前面要有0。

#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 main() {
    int n;
    cin >> n;
    while (n--) {
        int month, year, day;
        cin >> year >> month >> day;
        int leap = 0;
        if (year % 400 == 0 || year % 4 == 0 && year % 100 != 0)
            leap = 1;
        day++;
        if (day == a[leap][month] + 1) {
            month++;
            day = 1;
        }
        printf("%04d-%02d-%02d\n", year, month, day);
    }
    return 0;
}