请你编写一个程序来计算两个日期之间隔了多少天。
日期以字符串形式给出,格式为 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
年之间的有效日期。
解题思路
使用datetime
的话,只用一行即可。
import datetime
class Solution:
def daysBetweenDates(self, date1: str, date2: str) -> int:
return abs((datetime.datetime.strptime(date1, '%Y-%m-%d') - datetime.datetime.strptime(date2, '%Y-%m-%d')).days)
实际上这个问题和Leetcode 1154:一年中的第几天非常类似,本质都是闰年问题。思路非常简单,分别计算date1
和date2
到1971.1.1
有多少天,然后相减即可。
class Solution:
def daysBetweenDates(self, date1: str, date2: str) -> int:
days = [31,28,31,30,31,30,31,31,30,31,30,31]
def check(Y):
return Y % 4 == 0 and (Y % 100 != 0 or Y % 400 == 0)
def days_from_1971(date):
Y, M, D = list(map(int, date.split("-")))
for i in range(1971, Y):
D += 366 if check(i) else 365
return D + (M > 2 and check(Y)) + sum(days[:M - 1])
return abs(days_from_1971(date1) - days_from_1971(date2))
reference:
https://leetcode.com/problems/number-of-days-between-two-dates/discuss/517605/Similar-to-day-of-the-year
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!