[LeetCode 1360]日期之间隔几天

题目描述

请你编写一个程序来计算两个日期之间隔了多少天。

日期以字符串形式给出，格式为 YYYY-MM-DD，如示例所示。

示例1

输入：date1 = “2019-06-29”, date2 = “2019-06-30”
输出：1

示例2

输入：date1 = “2020-01-15”, date2 = “2019-12-31”
输出：15

提示

给定的日期是 1971 年到 2100 年之间的有效日期。

思路分析：

1.给定的是1971到2100年之间的有效日期，这样字符串处理会方便很多，因为年份长度都为4
2.以1970年为起点，计算当前日期到1970年1月1日的距离，计算两个日期之间的差值，就是他们相差的天数

代码

class Solution {
public:
    int convert(int year, int month, int day)
    {
    	//记录每月日期数量的数组
    	//一三五七八十腊，三十一天永不差（笑）
        int mth[13] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        int res = 0;
        for(int i = 1970; i < year; i++)
        {
        	//判断闰年
            if(i % 400 == 0 || i % 4 == 0 && i % 100 != 0) res += 366;
            else res += 365;
        }
        for(int i = 1; i < month; i++) res += mth[i];
        //如果今年是闰年，且月份大于2月，那么就要再+1
        if((year % 400 == 0 || year %4 == 0 && year % 100 != 0) && month > 2) res++;
        res += day;
        return res;
    }
    int daysBetweenDates(string date1, string date2) {
        int year1 = stoi(date1.substr(0, 4)), month1 = stoi(date1.substr(5, 2)), 
            day1 = stoi(date1.substr(8, 2));
        int year2 = stoi(date2.substr(0, 4)), month2 = stoi(date2.substr(5, 2)), 
            day2 = stoi(date2.substr(8, 2));

        return abs(convert(year1, month1, day1) - convert(year2, month2, day2));

    }
};