跳转至

729. 我的日程安排表 I

难度:中等

题目

实现一个 MyCalendar 类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。

当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订

日程可以用一对整数 startTimeendTime 表示,这里的时间是半开区间,即 [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 次。

Reference

题解

基本思路: 使用一个列表存放所有开始的时间和结束的时间。这个列表满足:(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)

评论