LeetCode: 729. 我的日程安排表 I¶
1、题目描述¶
实现一个 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)
示例 1:
MyCalendar(); MyCalendar.book(10, 20); // returns true MyCalendar.book(15, 25); // returns false MyCalendar.book(20, 30); // returns true 解释: 第一个日程安排可以添加到日历中. 第二个日程安排不能添加到日历中,因为时间 15 已经被第一个日程安排预定了。 第三个日程安排可以添加到日历中,因为第一个日程安排并不包含时间 20 。
说明:
每个测试用例,调用 MyCalendar.book 函数最多不超过 100次。
调用函数 MyCalendar.book(start, end)时, start 和 end 的取值范围为 [0, 10^9]。
2、解题思路¶
- 二分法查找相应的位置,判断是否存在集合交叉,不存在则插入并返回
True
from bisect import bisect_right class MyCalendar: def __init__(self): self.starts = [] self.ends = [] def book(self, start: int, end: int) -> bool: end = end - 1 if not self.starts: self.starts.append(start) self.ends.append(end) return True else: index = bisect_right(self.starts, end) if index == 0: self.starts.insert(0, start) self.ends.insert(0, end) return True else: if self.ends[index - 1] < start: self.starts.insert(index, start) self.ends.insert(index, end) return True else: return False # Your MyCalendar object will be instantiated and called as such: # obj = MyCalendar() # param_1 = obj.book(start,end)