-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path729. My Calendar I.py
38 lines (33 loc) · 1000 Bytes
/
729. 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
class TreeNode():
def __init__(self, s, e):
self.s = s
self.e = e
self.left = None
self.right = None
class MyCalendar:
def __init__(self):
self.root = None
def book(self, start: int, end: int) -> bool:
if not self.root:
self.root = TreeNode(start, end)
return True
else:
return self.insert(start, end, self.root)
def insert(self, s, e, node):
if s >= node.e:
if node.right:
return self.insert(s, e, node.right)
else:
node.right = TreeNode(s, e)
return True
elif e <= node.s:
if node.left:
return self.insert(s, e, node.left)
else:
node.left = TreeNode(s, e)
return True
else:
return False
# Your MyCalendar object will be instantiated and called as such:
# obj = MyCalendar()
# param_1 = obj.book(start,end)