【拒绝算法PUA】 732. 我的日程安排表 III

系列文章目录

【拒绝算法PUA】0x00-位运算
【拒绝算法PUA】0x01- 区间比较技巧
【拒绝算法PUA】0x02- 区间合并技巧
【拒绝算法PUA】0x03 - LeetCode 排序类型刷题
【拒绝算法PUA】LeetCode每日一题系列刷题汇总-2025年持续刷新中

C++刷题技巧总结：
[温习C/C++]0x04 刷题基础编码技巧

---

---

732. 我的日程安排表 III

链接

731. 我的日程安排表 II

题目

当 k 个日程存在一些非空交集时（即, k 个日程包含了一些相同时间），就会产生 k 次预订。

给你一些日程安排 [startTime, endTime) ，请你在每个日程安排添加后，返回一个整数 k ，表示所有先前日程安排会产生的最大 k 次预订。

实现一个 MyCalendarThree 类来存放你的日程安排，你可以一直添加新的日程安排。

MyCalendarThree() 初始化对象。
int book(int startTime, int endTime) 返回一个整数 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 <= startTime < endTime <= 109
每个测试用例，调用 book 函数最多不超过 400次

分析

输入：
[“MyCalendarThree”, “book”, “book”, “book”, “book”, “book”, “book”]
[[], [10, 20], [50, 60], [10, 40], [5, 15], [5, 10], [25, 55]]

绘制如下图：

通过扫描线法可以统计区间重叠问题。

从图中可以看到前三组数据有重叠， [5,10) 因为是右开区间，所以和前三组数据并无交集，因此当前用例正确结果为3。

解题方法1 (超出内存限制)

#include <iostream>
#include <vector>

using namespace std;

class MyCalendarThree {
private:
    vector<vector<int>> vec; // {{start, end, count}}
    int res = 0;
public:
    MyCalendarThree() {

    }

    int book(int startTime, int endTime) {
        vector<vector<int>> tmpVec;
        for (auto& data : vec) {
            int start = data[0];
            int end = data[1];
            int count = data[2];
            if (start < endTime && startTime < end) { // 说明有交集
                int left = max(start, startTime);
                int right = min(end, endTime);
                count += 1;
                res = max(res, count);
                tmpVec.push_back({left, right, count});
            }
        }
        vec.insert(vec.end(), tmpVec.begin(), tmpVec.end());
        vec.push_back({startTime, endTime, 1});
        res = max(res, 1);
        return res;
    }
};

解题方法2 (代码优化)

#include <iostream>
#include <vector>
#include <set>
#include <queue>

using namespace std;

class MyCalendarThree {
    multiset<pair<int, int>> mst;
public:
    MyCalendarThree() {}

    int book(int start, int end)
    {
        int ret = 0;
        mst.insert({start, end});

        priority_queue<int, vector<int>, greater<int>> q;

        for (auto &[s, e] : mst) {
            q.emplace(e);
            // int top = q.top();
            while (s >= q.top()) {
                q.pop();
            }
            ret = max(ret, (int)q.size());
        }
        return ret;
    }
};

解题方法3 (代码进一步优化)

#include <iostream>
#include <map>

using namespace std;

class MyCalendarThree {
    map<int, int> cnt;
public:
    MyCalendarThree() {}

    int book(int start, int end)
    {
        int ret = 0;
        int sum = 0;
        cnt[start]++; // 该时间有日程开始，日程数量+1
        cnt[end]--;  // 该时间有日程结束，日程数量-1
        for (auto& [key, freq] : cnt) {
            sum += freq;
            ret = max(sum, ret);
        }
        return ret;
    }
};