2409. 统计共同度过的日子数

提示

简单

77

相关企业

Alice 和 Bob 计划分别去罗马开会。

给你四个字符串 arriveAlice ,leaveAlice ,arriveBob 和 leaveBob 。Alice 会在日期 arriveAlice 到 leaveAlice 之间在城市里(日期为闭区间),而 Bob 在日期 arriveBob 到 leaveBob 之间在城市里(日期为闭区间)。每个字符串都包含 5 个字符,格式为 "MM-DD" ,对应着一个日期的月和日。

请你返回 Alice和 Bob 同时在罗马的天数。

你可以假设所有日期都在 同一个 自然年,而且 不是 闰年。每个月份的天数分别为:[31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31] 。

 

示例 1:

输入:arriveAlice = "08-15", leaveAlice = "08-18", arriveBob = "08-16", leaveBob = "08-19"
输出:3
解释:Alice 从 8 月 15 号到 8 月 18 号在罗马。Bob 从 8 月 16 号到 8 月 19 号在罗马,他们同时在罗马的日期为 8 月 16、17 和 18 号。所以答案为 3 。

示例 2:

输入:arriveAlice = "10-01", leaveAlice = "10-31", arriveBob = "11-01", leaveBob = "12-31"
输出:0
解释:Alice 和 Bob 没有同时在罗马的日子,所以我们返回 0 。

 

提示:

  • 所有日期的格式均为 "MM-DD" 。
  • Alice 和 Bob 的到达日期都 早于或等于 他们的离开日期。
  • 题目测试用例所给出的日期均为 非闰年 的有效日期。

Solution

class Solution:
    def countDaysTogether(self, arriveAlice: str, leaveAlice: str, arriveBob: str, leaveBob: str) -> int:
        days = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
        am1 = int(arriveAlice[:2])
        ad1 = int(arriveAlice[3:])
        bm1 = int(arriveBob[:2])
        bd1 = int(arriveBob[3:])
        am2 = int(leaveAlice[:2])
        ad2 = int(leaveAlice[3:])
        bm2 = int(leaveBob[:2])
        bd2 = int(leaveBob[3:])
        ads = sum(days[am1 - 1: am2 - 1]) - ad1 + ad2 + 1
        bds = sum(days[bm1 - 1: bm2 - 1]) - bd1 + bd2 + 1
        ints = 0
        if arriveAlice < arriveBob:
            ints = sum(days[am1 - 1: bm1 - 1]) - ad1 + bd1
            if ads > ints:
                return min(bds, ads - ints)
            else:
                return 0
        else:
            ints = sum(days[bm1 - 1: am1 - 1]) - bd1 + ad1
            if bds > ints:
                return min(ads, bds - ints)
            else:
                return 0

学习人家:

基本就是晚到和早退之间的差值。

class Solution:
    def countDaysTogether(self, arriveAlice: str, leaveAlice: str, arriveBob: str, leaveBob: str) -> int:
        months = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
        early_arrive = max(arriveAlice, arriveBob)
        late_leave = min(leaveAlice,leaveBob)

        if late_leave < early_arrive:
            return 0
        else:
            res = 0
            left_month = int(early_arrive.split('-')[0])
            left_day = int(early_arrive.split('-')[1])
            right_month = int(late_leave.split('-')[0])
            right_day = int(late_leave.split('-')[1])
            res += right_day - left_day + 1
            for i in range(left_month, right_month):
                res += months[i-1]
            return res