题目

实现一个 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解决方案，请移步力扣中等题解析

最后编辑于：2020.12.10 11:14:05