我的日程安排表I

2025-01-02

题目:

实现一个 MyCalendar 类来存放你的日程安排。如果要添加的日程安排不会造成 重复预订 ,则可以存储这个新的日程安排。

当两个日程安排有一些时间上的交叉时(例如两个日程安排都在同一时间内),就会产生 重复预订

日程可以用一对整数 startTimeendTime 表示,这里的时间是半开区间,即 [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);
 */