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

76 阅读2分钟

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

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

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

给你四个字符串 arriveAliceleaveAlicearriveBobleaveBob 。Alice 会在日期 arriveAliceleaveAlice 之间在城市里(日期为闭区间),而 Bob 在日期 arriveBobleaveBob 之间在城市里(日期为闭区间)。每个字符串都包含 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;
    }
};