-
Notifications
You must be signed in to change notification settings - Fork 1
/
0729-my-calendar-i.py
39 lines (33 loc) · 1.25 KB
/
0729-my-calendar-i.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from sortedcontainers import SortedDict
class MyCalendar:
def __init__(self):
self.dic = SortedDict()
# def book(self, start: int, end: int) -> bool:
# keys = self.dic.keys()
# if keys:
# r = self.dic.bisect_right(start)
# if r != 0:
# floorKey = self.dic[keys[r-1]]
# if floorKey > start: return False
# l = self.dic.bisect_left(start)
# if l != len(keys):
# ceilKey = keys[l]
# if end > ceilKey: return False
# self.dic[start] = end
# return True
def book(self, start: int, end: int) -> bool:
keys = self.dic.keys()
if keys:
i = self.dic.bisect_right(start)
if i != 0:
floorKey = self.dic[keys[i-1]]
if floorKey > start: return False
if i == len(keys) and start == keys[-1]: i -= 1 # do this equivalent to bisect_left() to save time
if i != len(keys):
ceilKey = keys[i]
if end > ceilKey: return False
self.dic[start] = end
return True
# Your MyCalendar object will be instantiated and called as such:
# obj = MyCalendar()
# param_1 = obj.book(start,end)