蓝桥杯省赛无忧 编程15 泡澡

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
struct Event {
    int time;
    int waterChange;
};
int main() {
    int N, W;
    cin >> N >> W;
    vector<Event> events;
    for (int i = 0; i < N; ++i) {
        int S, T, P;
        cin >> S >> T >> P;
        events.push_back({S, P});  // 在开始时间添加热水需求
        events.push_back({T, -P}); // 在结束时间减去热水需求
    }
    // 按时间对事件进行排序
    sort(events.begin(), events.end(), [](const Event &a, const Event &b) {
        return a.time < b.time;
    });

    int currentWater = 0; // 当前的热水需求

    // 遍历所有事件
    for (const auto &event : events) {
        currentWater += event.waterChange; // 更新热水需求
        if (currentWater > W) { // 如果超出了热水供应能力
            cout << "No" << endl;
            return 0;
        }
    }
    cout << "Yes" << endl; // 所有需求都能被满足
    return 0;
}

这个代码首先初始化两个整数变量N和W，用来读取人数和热水器的容量。然后定义一个结构体Event来存储每个事件的时间点和水量变化（需求增加或减少）。接着读取每个人的洗澡计划，将开始时间和结束时间作为事件存储到事件列表中。这里需要注意的是结束时间的水量变化是负数，表示热水需求的减少。

之后按照时间对所有事件进行排序，这样我们可以按照顺序遍历它们。在遍历的过程中，我们跟踪当前的热水需求，如果在任何时间点热水需求超过了热水器的供应能力W，我们就输出"No"并结束程序。如果最后所有的事件都被遍历过而没有超出供应能力，我们就输出"Yes"。这个方法的时间复杂度主要取决于排序操作，是O(N log N)。