实现一个 MyCalendar 类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。
当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订 。
日程可以用一对整数 start 和 end 表示,这里的时间是半开区间,即 [start, end), 实数 x 的范围为, start <= x < end 。
实现 MyCalendar 类:
MyCalendar() 初始化日历对象。
boolean book(int start, int end) 如果可以将日程安排成功添加到日历中而不会导致重复预订,返回 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 次。
构造一个二叉搜索树,如果插入区间的end比左边的节点还要小,并且左节点的left为空(说明此时的left已经没有能够重合的了),则遍历插入左节点的left,并返回为true,如果插入区间的start比右边的节点还要大,并且右节点的right为空(说明此时的right已经没有能够重合的了),则遍历插入右节点的right,并返回为true,否则说明区间有重合,返回为false
class TNode {
public:
TNode* left;
TNode* right;
int start;
int end;
TNode(int start, int end) :start(start), end(end), left(nullptr), right(nullptr) {}
};
class MyCalendar {
public:
MyCalendar() {
root = nullptr;
}
bool book(int start, int end) {
if (root == nullptr) {
root = new TNode(start, end);
return true;
}
TNode* cur = root;
while (true) {
if (end <= cur->start) {
if (cur->left == nullptr) {
cur->left = new TNode(start, end);
return true;
}
cur = cur->left;
}
else if (start >= cur->end) {
if (cur->right == nullptr) {
cur->right = new TNode(start, end);
return true;
}
cur = cur->right;
}
else {
return false;
}
}
}
private:
TNode* root;
};