【有序集合】我的日程安排表 I

题目

729. 我的日程安排表 I

---

实现一个 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 <= 10^9
• 每个测试用例，调用 book 方法的次数最多不超过 1000 次。

解题

方法一：有序集合

思路

维护一个树集，存储长度为2的数组表示一个日程区间，按照区间的起始升序排序。

每次添加新日程安排时：

• 利用树集的性质先找到在集合中应该排在当前日程(curr)后的第一个元素，如果该日程不为空且该日程起始比要添加日程的结束要大，则说明产生了重复预订，直接返回 false。
• 再找到应该排在当前日程(curr)前的第一个元素，如果该日程不为空且该日程结束比要添加日程的起始要大，则说明产生了重复预订，直接返回 false。
• 最后返回把 curr 加入集合(如果重复会返回 false)。

代码

class MyCalendar {
    private final TreeSet<Integer[]> set;

    public MyCalendar() {
        set = new TreeSet<>((a, b) -> a[0] - b[0]);
    }
    
    public boolean book(int start, int end) {
        Integer[] curr = {start, end};
        Integer[] ceiling = set.ceiling(curr);
        if (ceiling != null && ceiling[0] < end) return false;
        Integer[] floor = set.floor(curr);
        if (floor != null && floor[1] > start) return false;
        return set.add(curr);
    }
}

优化

class MyCalendar {
    private TreeSet<int[]> set;

    public MyCalendar() {
        set = new TreeSet<>((a, b) -> {
            if (a[1] <= b[0]) return -1;
            if (a[0] >= b[1]) return 1;
            return 0;
        });
    }
    
    public boolean book(int start, int end) {
        return set.add(new int[]{start, end});
    }
}