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 次。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/my-calendar-i
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

一开始往区间合并想了，结果做错了。

其实，可以发现，可以成功预定的区间都是互不重叠的，区间开始和结束时间都是升序的。

因此，对于新的区间，二分查找开始或者结束时间的插入位置，并判断和插入位置左右相邻区间是否有重叠即可。

代码

class MyCalendar {
    // map的key为区间开始时间，value为结束时间
    TreeMap<Integer, Integer> map;
    public MyCalendar() {
        map = new TreeMap<>();
    }
    public boolean book(int start, int end) {
        // 插入位置左边相邻区间的开始时间
        Integer left = map.floorKey(start);
        // 插入位置左边相邻区间的结束时间
        int e = 0;
        // 注意left可能为null
        if (left != null) {
            e = map.get(left);
        }
        // 插入位置右边相邻区间的开始时间
        Integer right = map.ceilingKey(start);
        // 需要同时满足下面两个条件就可以安排当前日程
        // 1.左边没有日程，或者和左边日程不重叠
        // 2.右边没有日程，或者和右边日程不重叠
        if ((left == null || start >= e) && (right == null || end <= right)) {
            map.put(start, end);
            return true;
        }
        return false;
    }
}
/**
 * Your MyCalendar object will be instantiated and called as such:
 * MyCalendar obj = new MyCalendar();
 * boolean param_1 = obj.book(start,end);
 */