目标
实现一个 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 <= 10^9
- 每个测试用例,调用 book 方法的次数最多不超过 1000 次。
提示:
- Store the events as a sorted list of intervals. If none of the events conflict, then the new event can be added.
思路
判断给定区间是否与已有区间相交,如果不相交将其加入已有区间。
最直接的想法是枚举每一个区间,判断是否相交,如果不相交则加入集合。判断区间 [a, b)
与 [c, d)
是否相交,可以固定一个区间,然后让另一个区间滑动,可以发现相交需要满足 d > a && c < b
,注意取等号是不相交的。
当然也可以使用二叉搜索树。
TreeSet 中查找特定元素的 API:
- ceiling 返回的是 大于等于 target 的最小元素/ null
- floor 返回的是 小于等于 target 的最大元素/ null
- higher 返回的是 大于 target 的最小元素 / null
- lower 返回的是 小于 target 的最大元素 / null
代码
/**
* @date 2025-01-02 9:58
*/
class MyCalendar {
TreeSet<int[]> ts = new TreeSet<>((a, b) -> a[0] - b[0]);
public MyCalendar() {
}
public boolean book(int startTime, int endTime) {
int[] interval = {startTime, endTime};
if (ts.isEmpty()) {
ts.add(interval);
return true;
}
int[] param = new int[]{endTime, 0};
// 查找 起点 大于等于 endTime 的 起点最小的区间,即要插入间隙的右边区间 right
int[] right = ts.ceiling(param);
// 如果 right 是第一个元素 或者 前面一个区间的右边界 end 小于等于 startTime,说明不相交
if (right == ts.first() || ts.lower(param)[1] <= startTime) {
ts.add(interval);
return true;
}
return false;
}
}