731. 我的日程安排2(Python)

2020-12-10  本文已影响0人  玖月晴

难度:★★★☆☆
类型:数组
方法:暴力

力扣链接请移步本题传送门
更多力扣中等题的解决方案请移步力扣中等题目录

题目

实现一个 MyCalendar 类来存放你的日程安排。如果要添加的时间内不会导致三重预订时,则可以存储这个新的日程安排。

MyCalendar 有一个 book(int start, int end)方法。它意味着在 start 到 end 时间内增加一个日程安排,注意,这里的时间是半开区间,即 [start, end), 实数 x 的范围为, start <= x < end。

当三个日程安排有一些时间上的交叉时(例如三个日程安排都在同一时间内),就会产生三重预订。

每次调用 MyCalendar.book方法时,如果可以将日程安排成功添加到日历中而不会导致三重预订,返回 true。否则,返回 false 并且不要将该日程安排添加到日历中。

请按照以下步骤调用MyCalendar 类: MyCalendar cal = new MyCalendar(); MyCalendar.book(start, end)

示例:

MyCalendar();
MyCalendar.book(10, 20); // returns true
MyCalendar.book(50, 60); // returns true
MyCalendar.book(10, 40); // returns true
MyCalendar.book(5, 15); // returns false
MyCalendar.book(5, 10); // returns true
MyCalendar.book(25, 55); // returns true
解释:
前两个日程安排可以添加至日历中。 第三个日程安排会导致双重预订,但可以添加至日历中。
第四个日程安排活动(5,15)不能添加至日历中,因为它会导致三重预订。
第五个日程安排(5,10)可以添加至日历中,因为它未使用已经双重预订的时间10。
第六个日程安排(25,55)可以添加至日历中,因为时间 [25,40] 将和第三个日程安排双重预订;
时间 [40,50] 将单独预订,时间 [50,55)将和第二个日程安排双重预订。

提示:

每个测试用例,调用 MyCalendar.book 函数最多不超过 1000次。
调用函数 MyCalendar.book(start, end)时, start 和 end 的取值范围为 [0, 10^9]。

解答

这里参考官网解答。

跟上一道题不同的是,由于时间段之间可以存在重合,我们需要准备两个列表,其中一个是calendar,用来存储各个时间段,另一个是overlaps,用例存储重合时间段的公共区域。如果新加入的时间段与日程表中任意时间段存在重合,我们就把重合的时间段添加到overlaps列表中。因此,判断新加入的时间段是否能够加入到日程表中,只需要判断overlaps列表中是否存在与新加入的时间段重合的时间段即可。

这里的重点是,如何判断两个时间段之间存在公共区域。我们可以从相反的角度来考虑。两个时间段[current_start, current_end]与[start, end]不重合的条件是end <= current_start or current_end <= start,那么取其相反即为重合的条件。根据这个,我们就可以实现时间段与时间段之间是否重合的判断,判断新加入的时间段是否与overlaps列表中或者calendar列表中的时间段存在重合。

注意overlaps中时间段的获取,应该是新加入的时间段与calendar时间段中的重合区域,即(max(start, current_start), min(end, current_end)),这有点像视觉算法中计算IOU的公共方框。

class MyCalendarTwo:
    def __init__(self):
        self.calendar = []
        self.overlaps = []

    def book(self, start, end):
        for current_start, current_end in self.overlaps:
            if not (end <= current_start or current_end <= start):
                return False
        for current_start, current_end in self.calendar:
            if not (end <= current_start or current_end <= start):
                self.overlaps.append((max(start, current_start), min(end, current_end)))
        
        self.calendar.append((start, end))
        return True

如有疑问或建议,欢迎评论区留言~

有关更多力扣中等题的python解决方案,请移步力扣中等题解析

上一篇下一篇

猜你喜欢

热点阅读