题目链接,描述
https://www.lintcode.com/problem/1063
java
实现一个MyCalendarThree 来储存你的时间。一个新的事件 总是 可以被加入。
你的类会有一种方法:book(int start, int end)。 正式的说,这代表在一个半开区间 [start, end) 上进行预订,实数x 的范围即 start <= x < end。
当K个事件有一个非空交集的时候,一个K预订将会发生。(即,某一个时刻对于K个事件是共用的)
对于每一个对于方法 MyCalendar.book的调用,返回一个整数K,代表日历中存在K预订的最大的整数。
你的类将会这样被调用: MyCalendarThree cal = new MyCalendarThree(); MyCalendarThree.book(start, end)。
每个测试用例将会调用 MyCalendarThree.book 至多400次 。
在调用MyCalendarThree.book(start, end)时, start 和 end是在[0, 10^9]内的整数。
样例
样例1
输入:
MyCalendarThree()
book(10,20)
book(50,60)
book(10,40)
book(5,15)
book(5,10)
book(25,55)
输出: [1,1,2,3,3,3]
说明:
前两个可以被预订的事件是不相交的,所以说最大的K预订是1预订。
第三个事件[10, 40)和第一个事件相交,最大的K预订是2预订。
剩下的事件导致最大的K预订只是3预订。
注意到最后一个事件在本地导致2预订,但是答案依然是3预订
因为比如, [10, 20), [10, 40), and [5, 15)依然是三重预订。
样例2
输入:
MyCalendarThree()
book(1,2)
book(1,2)
book(2,3)
输出: [1,2,2]
思路
前置知识:线段树
参考代码
java
class MyCalendarThree {
SegmentNode root; //线段树根节点
public MyCalendarThree() {
root = build(0,100000);
}
public int book(int start, int end) {
for (int i = start; i < end; i++) {
update(root, i, 1);
}
return root.max;
}
static class SegmentNode { //线段树节点定义
int start, end, max;
SegmentNode left, right;
public SegmentNode(int s, int e) {
start = s;
end = e;
}
}
//根据开始,结束标志,创建线段树
public static SegmentNode build(int start, int end) {
if (start > end) return null;
if (start == end)
return new SegmentNode(start, end);
SegmentNode root = new SegmentNode(start, end);
int mid = start + (end - start) / 2;
root.left = build(start, mid);
root.right = build(mid + 1, end);
return root;
}
//更新线段树
public static void update(SegmentNode root, int index, int val) {
if (root.start == root.end && root.start == index) {
root.max += val;
return;
}
int mid = (root.start + root.end) / 2;
if (index <= mid && index >= root.start) {
update(root.left, index, val);
}
if (index > mid && index <= root.end) {
update(root.right, index, val);
}
root.max = Math.max(root.left.max, root.right.max);
}
}
/**
* Your MyCalendarThree object will be instantiated and called as such:
* MyCalendarThree obj = new MyCalendarThree();
* int param_1 = obj.book(start,end);
*/