题目描述
当 k
个日程安排有一些时间上的交叉时(例如 k
个日程安排都在同一时间内),就会产生 k
次预订。
给你一些日程安排 [start, end)
,请你在每个日程安排添加后,返回一个整数 k
,表示所有先前日程安排会产生的最大 k
次预订。
实现一个 MyCalendarThree
类来存放你的日程安排,你可以一直添加新的日程安排。
MyCalendarThree()
初始化对象。int book(int start, int end)
返回一个整数k
,表示日历中存在的k
次预订的最大值。
示例:
输入:
["MyCalendarThree", "book", "book", "book", "book", "book", "book"]
[[], [10, 20], [50, 60], [10, 40], [5, 15], [5, 10], [25, 55]]
输出:
[null, 1, 1, 2, 3, 3, 3]
解释:
MyCalendarThree myCalendarThree = new MyCalendarThree();
myCalendarThree.book(10, 20); // 返回 1 ,第一个日程安排可以预订并且不存在相交,所以最大 k 次预订是 1 次预订。
myCalendarThree.book(50, 60); // 返回 1 ,第二个日程安排可以预订并且不存在相交,所以最大 k 次预订是 1 次预订。
myCalendarThree.book(10, 40); // 返回 2 ,第三个日程安排 [10, 40) 与第一个日程安排相交,所以最大 k 次预订是 2 次预订。
myCalendarThree.book(5, 15); // 返回 3 ,剩下的日程安排的最大 k 次预订是 3 次预订。
myCalendarThree.book(5, 10); // 返回 3
myCalendarThree.book(25, 55); // 返回 3
提示:
0 <= start < end <= 10^9
- 每个测试用例,调用
book
函数最多不超过400
次
看看评论区的解法,有差分数组,还有线段树,于是我选择果断看差分数组(线段树实在看不懂)。
于是我学习了一下差分数组,感觉就是和前缀和相反的操作,前缀和是计算到当前为止的数组元素之和,差分数组是计算相邻数组元素之差,也可以看成是一种作差的递推关系式衍生的新数组。
差分数组的产生应该是为了满足快速更新数组某个区间的值的需求,正常的形式是[start,end),从start开始增量,到end为止减去增量。
class MyCalendarThree {
private TreeMap<Integer, Integer> cnt;
public MyCalendarThree() {
cnt = new TreeMap<Integer, Integer>();
}
public int book(int start, int end) {
int ans = 0;
int maxBook = 0;
cnt.put(start, cnt.getOrDefault(start, 0) + 1);
cnt.put(end, cnt.getOrDefault(end, 0) - 1);
for (Map.Entry<Integer, Integer> entry : cnt.entrySet()) {
int freq = entry.getValue();
maxBook += freq;
ans = Math.max(maxBook, ans);
}
return ans;
}
}
学习到一个新的知识:差分数组。以后的练习中继续熟悉,深入吧。