【LeetCode 每日一题】2409. 统计共同度过的日子数

202 阅读2分钟

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

难度:简单

时间:2023/04/17


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 的到达日期都 早于或等于 他们的离开日期。

    • 题目测试用例所给出的日期均为 非闰年 的有效日期。

解题思路:模拟

  • 先把日期转换成天数; 设alice的时间是[day0,day1], bob的时间是[day2,day3];
  • 按照天数排序得到situation(如[day2,day3,day0,day1]);
  • 可以将所有情况分为两个大类:
    • 1、没交集:[day0,day1,day2,day3]或者[day2,day3,day0,day1](但需要特判situation[0]==situation[1]的情况)。
    • 2、有交集:此时答案就是:situation[2]的天数-situation[1]的天数+1
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]     
        
        def date2day(s: str) -> int:
            m, d = map(int, s.split('-'))
            return sum(days[:m-1]) + d
        
        alice_bob = [date2day(arriveAlice), date2day(leaveAlice), date2day(arriveBob), date2day(leaveBob)]
​
        raw_idx = [0, 1, 2, 3]
​
        temp_list = sorted(zip(alice_bob, raw_idx))
        _, situation = zip(*temp_list)
        if situation in [(0,1,2,3),(2,3,0,1)]:
            if alice_bob[situation[2]] == alice_bob[situation[1]]:
                return 1
            else:
                return 0
        else:
            return alice_bob[situation[2]] - alice_bob[situation[1]] + 1
​
            
        
​