LeetCode題解(1360):日期之間隔幾天(Python)

題目:原題鏈接(簡單)

解法 時間複雜度 空間複雜度 執行用時
Ans 1 (Python) 36ms (92.68%)
Ans 2 (Python) 36ms (92.68%)
Ans 3 (Python)

LeetCode的Python執行用時隨緣,只要時間複雜度沒有明顯差異,執行用時一般都在同一個量級,僅作參考意義。

解法一(樸素算法):

def daysBetweenDates(self, date1: str, date2: str) -> int:
    if date1 > date2:
        date1, date2 = date2, date1
    y1, m1, d1 = date1.split("-")
    y1, m1, d1 = int(y1), int(m1), int(d1)
    y2, m2, d2 = date2.split("-")
    y2, m2, d2 = int(y2), int(m2), int(d2)

    months = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    months_leap = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

    ans = 0

    # 統計整年
    for y in range(y1 + 1, y2):
        if y % 4 == 0 and (y % 100 != 0 or y % 400 == 0):
            ans += 366
        else:
            ans += 365

    # 統計整月
    months1 = months_leap if y1 % 4 == 0 and (y1 % 100 != 0 or y1 % 400 == 0) else months
    months2 = months_leap if y1 % 4 == 0 and (y1 % 100 != 0 or y1 % 400 == 0) else months
    if y1 != y2:
        ans += sum(months1[m1:])
        ans += sum(months2[:m2 - 1])

    # 統計剩下的日期
    if y1 != y2 or m1 != m2:
        ans += months1[m1 - 1] - d1
        ans += d2
    else:
        ans += d2 - d1

    return ans

解法二(使用datetime):

def daysBetweenDates(self, date1: str, date2: str) -> int:
    if date1 > date2:
        date1, date2 = date2, date1
    y1, m1, d1 = date1.split("-")
    y1, m1, d1 = int(y1), int(m1), int(d1)
    y2, m2, d2 = date2.split("-")
    y2, m2, d2 = int(y2), int(m2), int(d2)
    date1 = datetime.datetime(y1, m1, d1)
    date2 = datetime.datetime(y2, m2, d2)
    return (date2 - date1).days
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章