data_structure_algorithm/algorithm/code/get_next_second.cpp at master · T-tssxuan/data_structure_algorithm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include <iostream>
#include <string>
#include <vector>
#include <iomanip>
using namespace std;

/**
 * Describe: Give you a time string with the format: YYYY-MM-DD HH:MM:SS get the
 * next second of the moment. Times of 400 is leap year, times of 100 is not
 * leap year, time of 4 year is leap year.
 */

class Solution {
public:
    void getNextSecond(string str) {
        vector<int> info = strToInfo(str);
        calculateNext(info);
        outputResult(info);
    }

    // calculate the next HH:MM:SS
    void calculateNext(vector<int>& info) {
        if (info[5] == 59) {
            info[5] = 0;
            if (info[4] == 59) {
                info[4] = 0;
                if (info[3] == 23) {
                    info[3] = 0;
                    calculateDMY(info);
                } else {
                    info[3]++;
                }
            } else {
                info[4]++;
            }
        } else {
            info[5]++;
        }
    }

    // calculate the next YYYY:MM:DD
    void calculateDMY(vector<int>& info) {
        bool is_leap_year = isLeapYear(info[0]);
        int months[13] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        int carry = 0;
        if (info[1] != 2) {
            if (info[2] == months[info[1]]) {
                info[2] = 1;
                carry = 1;
            } else {
                info[2]++;
            }
        } else {
            if (info[2] == 28 && is_leap_year == false) {
                info[2] = 1;
                carry = 1;
            } else if (is_leap_year == true && info[2] == 29) {
                info[2] = 1;
                carry = 1;
            } else {
                info[2]++;
            }
        }
        if (carry == 1) {
            if (info[1] == 12) {
                info[1] = 1;
                info[0]++;
            } else {
                info[1]++;
            }
        }
    }

    // determain whether is leap year
    bool isLeapYear(int year) {
        if (year % 400 == 0) {
            return true;
        } else if (year % 100 == 0) {
            return false;
        } else if (year % 4 == 0) {
            return true;
        } else {
            return false;
        }
    }

    // output the string to time
    vector<int> strToInfo(string& str) {
        vector<int> info(6, 0);
        info[0] = stringToInt(str.substr(0, 4));
        info[1] = stringToInt(str.substr(5, 2));
        info[2] = stringToInt(str.substr(8, 2));
        info[3] = stringToInt(str.substr(11, 2));
        info[4] = stringToInt(str.substr(14, 2));
        info[5] = stringToInt(str.substr(17, 2));
        return info;
    }

    // convert string to time
    void outputResult(vector<int>& info) {
        cout << setw(4) << setfill('0') << info[0] << "/"
            << setw(2) << info[1] << "/"
            << setw(2) << info[2] << " "
            << setw(2) << info[3] << ":"
            << setw(2) << info[4] << ":"
            << setw(2) << info[5] << endl;
    }

    // convert string to number
    int stringToInt(string str) {
        int re = 0;
        for (int i = 0; i < str.size(); i++) {
            re = re * 10 + str[i] - '0';
        }
        return re;
    }
};

int main() {
    Solution so;
    string str;
    getline(cin, str);
    so.getNextSecond(str);
    return 0;
}