剑指 Offer II 058. 日程表

剑指 Offer II 058. 日程表

原始题目链接：https://leetcode.cn/problems/fi9suh/
与题729. 我的日程安排表 I 相同https://leetcode.cn/problems/my-calendar-i/

请实现一个 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”,“book”,“book”,“book”]
[[],[10,20],[15,25],[20,30]]
输出: [null,true,false,true]
解释:
MyCalendar myCalendar = new MyCalendar();
MyCalendar.book(10, 20); // returns true
MyCalendar.book(15, 25); // returns false ，第二个日程安排不能添加到日历中，因为时间 15 已经被第一个日程安排预定了
MyCalendar.book(20, 30); // returns true ，第三个日程安排可以添加到日历中，因为第一个日程安排并不包含时间 20

解题思路：

使用有序集合，再利用二分查找，判断条件。

代码实现：

from sortedcontainers import SortedList
import bisect

class MyCalendar:

    def __init__(self):
        # 初始化一个有序数组，存储时间对
        self.arr = SortedList()


    def book(self, start: int, end: int) -> bool:
        # 使用二分查找要插入的时间对在有序数组的位置
        idx = bisect.bisect_left(self.arr, (start, end))
        # 还没插入新的时间对之前，如果索引大于0并且要插入的位置之前的end是大于要插入时间对的start
        # 就是前一个时间对还没有结束，就开始新的日程是不满足条件的
        if idx > 0 and self.arr[idx - 1][1] > start:
            return False
        # 还没插入新的时间对之前，如果索引小于有序数组的长度，并且要插入的位置的（不是新的时间对是已存在的时间对）
        # 时间对的开时间早于新的时间对结束时间，因为新插入的时间对在要插入时间对位置的左边
        if idx < len(self.arr) and self.arr[idx][0] < end:
            return False
        
        # 加入新的时间对
        self.arr.add((start, end))

        return True



# Your MyCalendar object will be instantiated and called as such:
# obj = MyCalendar()
# param_1 = obj.book(start,end)

参考文献：
https://leetcode.cn/problems/fi9suh/solution/sortedlist-by-sanjin09-qdik/