729. 我的日程安排表 I¶
难度:中等
题目¶
实现一个 MyCalendar
类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。
当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订 。
日程可以用一对整数 startTime
和 endTime
表示,这里的时间是半开区间,即 [startTime, endTime)
, 实数 x
的范围为, startTime <= x < endTime
。
实现 MyCalendar
类:
MyCalendar()
初始化日历对象。boolean book(int startTime, int endTime)
如果可以将日程安排成功添加到日历中而不会导致重复预订,返回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
次。
题解¶
基本思路: 使用一个列表存放所有开始的时间和结束的时间。这个列表满足:(1) 列表单调非减;(2) 偶数下标的时间是开始时间,奇数下标的时间是结束时间。
在把新的时间段加入到列表时,两个时间加入的位置需要满足:(1) 两个位置之间需要没有其他元素,否则这个区间内包含了另外一个区间的开始或结束,即包含重叠。(2) 插入位置之前的元素必须是奇数下标(对应到一个区间的结束时间),否则插入的位置是一个已经记录的时间段的中间。
bisect
库提供了二分查找和二分插入的相关函数。由于时间区间为左包含,因此,在检查开始时间时,插入位置需要是右侧插入的位置;同理,检查结束时间时,插入位置需要是左侧插入的位置。
代码如下:
from typing import List, Optional, Tuple
import bisect
class MyCalendar:
def __init__(self):
self.times: List[int] = []
def _check(self, startTime: int, endTime: int) -> Optional[int]:
assert not (len(self.times) % 2)
x = bisect.bisect_right(self.times, startTime)
y = bisect.bisect_left(self.times, endTime)
if x != y:
return None
if x % 2:
return None
return x
def book(self, startTime: int, endTime: int) -> bool:
result = self._check(startTime, endTime)
if result is None:
return False
bisect.insort_left(self.times, endTime)
bisect.insort_right(self.times, startTime)
return True
# Your MyCalendar object will be instantiated and called as such:
# obj = MyCalendar()
# param_1 = obj.book(startTime,endTime)