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