一、题目
实现一个 MyCalendar
类来存放你的日程安排。如果要添加的时间内不会导致三重预订时,则可以存储这个新的日程安排。
MyCalendar
有一个 book(int start, int end)
方法。它意味着在 start
到 end
时间内增加一个日程安排,注意,这里的时间是半开区间,即 [start, end)
, 实数 x
的范围为, start <= x < end
。
当三个日程安排有一些时间上的交叉时(例如三个日程安排都在同一时间内),就会产生三重预订。
每次调用 MyCalendar.book
方法时,如果可以将日程安排成功添加到日历中而不会导致三重预订,返回 true
。否则,返回 false
并且不要将该日程安排添加到日历中。
请按照以下步骤调用MyCalendar
类: MyCalendar cal = new MyCalendar(); MyCalendar.book(start, end)
二、示例
示例一:
MyCalendar();
MyCalendar.book(10, 20); // returns true
MyCalendar.book(50, 60); // returns true
MyCalendar.book(10, 40); // returns true
MyCalendar.book(5, 15); // returns false
MyCalendar.book(5, 10); // returns true
MyCalendar.book(25, 55); // returns true
【解释】
- 前两个日程安排可以添加至日历中。 第三个日程安排会导致双重预订,但可以添加至日历中。
- 第四个日程安排活动(5,15)不能添加至日历中,因为它会导致三重预订。
- 第五个日程安排(5,10)可以添加至日历中,因为它未使用已经双重预订的时间10。
- 第六个日程安排(25,55)可以添加至日历中,因为时间 [25,40] 将和第三个日程安排双重预订;
- 时间 [40,50] 将单独预订,时间 [50,55)将和第二个日程安排双重预订。
提示:
每个测试用例,调用 MyCalendar.book
函数最多不超过 1000
次。
调用函数 MyCalendar.book(start, end)
时, start
和 end
的取值范围为 [0, 10^9]
。
三、解题思路
线段树
解决的是区间和的问题,且该区间会被修改。所以线段树主要实现两个方法:【求区间和】
和【修改区间】
,且时间复杂度均为 **O(logn)**。
始终记住一句话:线段树的每个节点代表一个区间。
既然需要以节点方式展示,并且需要以懒惰标记方式存储值,所以,针对每个节点的数据结构,如下所示:
/** 线段树节点结构 */
class Node {
Node left; // 左子节点
Node right; // 右子节点
int val = 0; // 当前节点值
int add = 0; // 懒惰标记值
}
什么叫懒惰标记呢? 可以做个比喻。比如我们要将区间[0, 999]这范围内的1000个节点的val值都修改为1,那么我们可以选择把每个叶子节点和区间节点都修改为1,但是这样做,效率会比较低。那么,还有一种方式是,在区间节点[0, 999]的val值修改为1,并且将该节点的add修改为1。那么当遍历到它的子节点的时候,我们再通过节点存储的add值“下移”给子节点,即:[子节点].val += [区间节点].add
。
我们以安排(10,20)
为例,由于默认节点val值为0,所以相关的区间节点都会修改为1。如下是其转换线段树的逻辑:
我们以安排(50,60)
为例,由于默认节点val值为0,所以相关的区间节点都会修改为1。如下是其转换线段树的逻辑:
我们以安排(10,40)
为例,由于(10,20)已经在上面步骤中被赋值为1了,并且这部分区间是有重叠的,所以相关的区间节点都会修改为2。如下是其转换线段树的逻辑:
我们以安排(5,15)
为例,由于(10,20)已经在上面步骤中被赋值为2了,并且这部分区间是有重叠的,所以执行查询方法的时候,查询出在(5,15)中已经有节点为2。那么,如果返回2这个值,则整个插入方法为false,即:无法添加此日程安排。如下是其转换线段树的逻辑:
四、代码实现
4.1> 实现1:线段树 + 懒惰标记
package com.muse;
/**
* 731. 我的日程安排表 II
*/
class MyCalendarTwo {
private static int START = 0; // 开始区间0
private static int END = (int) 1e9; // 结束区间10^9
private static int searchStart = 0; // 待查询的开始区间值
private static int searchEnd = 0; // 待查询的结束区间值
private Node root = new Node(); // 创建根节点
/** 线段树节点结构 */
class Node {
Node left; // 左子节点
Node right; // 右子节点
int val = 0; // 当前节点值
int add = 0; // 懒惰标记值
}
public MyCalendarTwo() {
}
public boolean book(int start, int end) {
searchStart = start;
searchEnd = end - 1; // 因为是[start, end),所以end要减1
if (query(root, START, END) == 2) {
return false;
}
update(root, START, END, 1);
return true;
}
/** 查询某一区间内最大val值 */
public int query(Node node, int start, int end) {
// 如果待查询的范围包含了node的范围,则直接返回node.val值
if (searchStart <= start && end <= searchEnd) {
return node.val;
}
// 如果没有左右子节点,则创建
pushDown(node);
int result = 0;
int middle = (end + start) >> 1; // 取start到end的中间点
if (searchStart <= middle) {
result = query(node.left, start, middle);
}
if (middle < searchEnd) {
result = Math.max(result, query(node.right, middle + 1, end));
}
return result;
}
/** 更新线段树中的相关节点val值 */
public void update(Node node, int start, int end, int value) {
if (searchStart <= start && end <= searchEnd) {
node.val += value;
node.add += value;
return;
}
// 如果没有左右子节点,则创建
pushDown(node);
int middle = (end + start) >> 1; // 取start到end的中间点
if (searchStart <= middle) {
update(node.left, start, middle, value);
}
if (middle < searchEnd) {
update(node.right, middle + 1, end, value);
}
pushUp(node);
}
/** 左右子节点val值的最大值就是父节点val值 */
public void pushUp(Node node) {
node.val = Math.max(node.left.val, node.right.val);
}
/** 创建左右子节点,并且下移懒惰标记值 */
public void pushDown(Node node) {
if (node.left == null) {
node.left = new Node();
}
if (node.right == null) {
node.right = new Node();
}
// 如果懒惰标记值add等于0,则不需要执行下面的【下移懒惰标记值】操作了,直接返回
if (node.add == 0) {
return;
}
// 【下移懒惰标记值】操作
node.left.val += node.add;
node.left.add += node.add;
node.right.val += node.add;
node.right.add += node.add;
node.add = 0;
}
}
今天的文章内容就这些了:
写作不易,笔者几个小时甚至数天完成的一篇文章,只愿换来您几秒钟的点赞&分享。
更多技术干货,欢迎大家关注公众号“爪哇缪斯”(^o^)/~ 「干货分享,每周更新」
往期精选
多图详解阻塞队列——SynchronousQueuehttps://blog.csdn.net/qq_26470817/article/details/125734927?spm=1001.2014.3001.5501ThreadLocal源码解析https://blog.csdn.net/qq_26470817/article/details/124993311?spm=1001.2014.3001.5501JVM原理与实战https://blog.csdn.net/qq_26470817/article/details/124890946?spm=1001.2014.3001.5501万字图文——HashMap源码解析(包含红黑树)https://blog.csdn.net/qq_26470817/article/details/121890240?spm=1001.2014.3001.5501
题目来源:力扣(LeetCode)