当 k
个日程存在一些非空交集时(即, k
个日程包含了一些相同时间),就会产生 k
次预订。
给你一些日程安排 [startTime, endTime)
,请你在每个日程安排添加后,返回一个整数 k
,表示所有先前日程安排会产生的最大 k
次预订。
实现一个 MyCalendarThree
类来存放你的日程安排,你可以一直添加新的日程安排。
MyCalendarThree()
初始化对象。int book(int startTime, int endTime)
返回一个整数k
,表示日历中存在的k
次预订的最大值。
思路
用差分数组每当我们预定一个新的日程安排 [start,end),在 start 计数 cnt[start] 加 1,表示从 start 预定的数目加 1;从 end 计数 cnt[end] 减 1,表示从 end 开始预定的数目减 1。然后遍历计数,求出最大的预定数目即可。
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;
}
}