力扣——2409. 统计共同度过的日子数
2409. 统计共同度过的日子数
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 的到达日期都 早于或等于 他们的离开日期。
- 题目测试用例所给出的日期均为 非闰年 的有效日期。
问题解析
这题的主要难点就在于每个月天数的不同导致计算天数比较麻烦。
但是我们可以转换一下,从几月几日改变为当前日期是今年的第几天,比如1月1日是今年的第1天。这样就可以把开始日期和离开日期变成一个区间了。
然后我们合并两个区间,求他们合并后的区间的宽度,就可以知道他们同时在几天了。
AC代码
class Solution {
public:
int get_day(string s,vector<int>&day)
{
int x=(s[0]-'0')*10+(s[1]-'0'),y=(s[3]-'0')*10+(s[4]-'0');
int sum=0;
for(int i=1;i<x;i++)sum+=day[i];
return sum+y;
}
int countDaysTogether(string arriveAlice, string leaveAlice, string arriveBob, string leaveBob) {
int l=0,r=0,l2=0,r2=0;
vector<int>day={0,31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
l=get_day(arriveAlice,day);
r=get_day(leaveAlice,day);
l2=get_day(arriveBob,day);
r2=get_day(leaveBob,day);
int x=max(l,l2),y=min(r,r2);
if(x>y)return 0;
return y-x+1;
}
};