题目:
实现一个 MyCalendar
类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。
当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订 。
日程可以用一对整数 startTime
和 endTime
表示,这里的时间是半开区间,即 [startTime, endTime)
, 实数 x
的范围为, startTime <= x < endTime
。
实现 MyCalendar
类:
MyCalendar()
初始化日历对象。boolean book(int startTime, int endTime)
如果可以将日程安排成功添加到日历中而不会导致重复预订,返回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 <= 109
- 每个测试用例,调用
book
方法的次数最多不超过1000
次。
思路:
用优先级队列存储[startTime, endTime),面对要新加入的日程时,判断是否和队列中的每一个元素有交集,都没有就说明可以加入,返回true
代码:
class MyCalendar {
public:
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
MyCalendar() {
}
bool book(int startTime, int endTime) {
vector<pair<int, int>> tmp;
if (pq.empty()) {
pq.push(make_pair(startTime, endTime));
return true;
}
bool flag = true;
while (!pq.empty()) {
auto top = pq.top();
if (startTime >= top.second || endTime <= top.first) {
tmp.push_back(top);
pq.pop();
continue;
}
flag = false;
break;
}
for (auto val : tmp) {
pq.push(val);
}
if (flag) {
pq.push(make_pair(startTime, endTime));
return true;
}
return false;
}
};
/**
* Your MyCalendar object will be instantiated and called as such:
* MyCalendar* obj = new MyCalendar();
* bool param_1 = obj->book(startTime,endTime);
*/