729. 我的日程安排表 I

Difficulty
Medium
Tags
二分查找
TreeMap
URL
https://leetcode.cn/problems/my-calendar-i/
Star
实现一个 MyCalendar 类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ，则可以存储这个新的日程安排。
当两个日程安排有一些时间上的交叉时（例如两个日程安排都在同一时间内），就会产生 重复预订 。
日程可以用一对整数 start 和 end 表示，这里的时间是半开区间，即 [start, end), 实数 x 的范围为， start <= x < end 。
实现 MyCalendar 类：
  • MyCalendar() 初始化日历对象。
  • boolean book(int start, int end) 如果可以将日程安排成功添加到日历中而不会导致重复预订，返回 true 。否则，返回 false 并且不要将该日程安排添加到日历中。
示例：
输入：
["MyCalendar", "book", "book", "book"]
[[], [10, 20], [15, 25], [20, 30]]
输出：
[null, true, false, true]

解释：
MyCalendar myCalendar = new MyCalendar();
myCalendar.book(10, 20); // return True
myCalendar.book(15, 25); // return False ，这个日程安排不能添加到日历中，因为时间 15 已经被另一个日程安排预订了。
myCalendar.book(20, 30); // return True ，这个日程安排可以添加到日历中，因为第一个日程安排预订的每个时间都小于 20 ，且不包含时间 20 。
提示：
  • 0 <= start < end <= 109
  • 每个测试用例，调用 book 方法的次数最多不超过 1000 次。
通过次数17,036提交次数30,846

法1 二分查找

思路
维护一个有序的列表（只需要保证起始时间有序即可）
然后每次需要插入时，都通过二分查找找到插入位置，然后判断是否合法。
题解
class MyCalendar:

    def __init__(self):
        self.arr = []

    def book(self, start: int, end: int) -> bool:
        # 找大于 start 的最小值 下标
        l, r = 0, len(self.arr)-1
        while l <= r:
            m = l + (r-l) // 2
            if self.arr[m][0] < start:
                l = m + 1
            elif self.arr[m][0] >= start:
                r = m - 1
        # 只需要考虑一下两种情况：
        # 1. 假设插入后，和后面的是否冲突
        # 2. 假设插入后，和前面的是否冲突
        if l > 0 and start < self.arr[l-1][1]:
            return False

        if l < len(self.arr) and end > self.arr[l][0]:
            return False

        self.arr.insert(l, [start, end])
        return True

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