503. 借教室(AcWing)，使用差分和二分

本题具体解释请看代码注释

在大学期间，经常需要租借教室。

大到院系举办活动，小到学习小组自习讨论，都需要向学校申请借教室。

教室的大小功能不同，借教室人的身份不同，借教室的手续也不一样。 

面对海量租借教室的信息，我们自然希望编程解决这个问题。

我们需要处理接下来 n天的借教室信息，其中第 i天学校有 ri 个教室可供租借。

共有 m 份订单，每份订单用三个正整数描述，分别为 dj,sj,tj，表示某租借者需要从第 sj 天到第 tj 天租借教室（包括第 sj天和第 tj天），每天需要租借 dj 个教室。 

我们假定，租借者对教室的大小、地点没有要求。

即对于每份订单，我们只需要每天提供 dj 个教室，而它们具体是哪些教室，每天是否是相同的教室则不用考虑。 

借教室的原则是先到先得，也就是说我们要按照订单的先后顺序依次为每份订单分配教室。

如果在分配的过程中遇到一份订单无法完全满足，则需要停止教室的分配，通知当前申请人修改订单。

这里的无法满足指从第 sj 天到第 tj天中有至少一天剩余的教室数量不足 dj 个。 

现在我们需要知道，是否会有订单无法完全满足。

如果有，需要通知哪一个申请人修改订单。

输入格式

第一行包含两个正整数 n,m，表示天数和订单的数量。 

第二行包含 n 个正整数，其中第 i 个数为 ri，表示第 i 天可用于租借的教室数量。 

接下来有 m行，每行包含三个正整数 dj,sj,tj表示租借的数量，租借开始、结束分别在第几天。 

每行相邻的两个数之间均用一个空格隔开。

天数与订单均用从 11 开始的整数编号。

输出格式

如果所有订单均可满足，则输出只有一行，包含一个整数 00。

否则（订单无法完全满足）输出两行，第一行输出一个负整数 −1−1，第二行输出需要修改订单的申请人编号。

输入样例：

4 3
2 5 4 3
2 1 3
3 2 4
4 2 4

输出样例：

-1
2

 

#include <vector>
#include <unordered_map>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <string>
#include <deque>
#include <map>
#include <cstdio>
#include <queue>
#include <climits>
#include <iostream>
#include <cstring>
#include <algorithm>

/*
 * 本轮算法是使用二分查找具体哪个订单数是第一个需要修改的订单
 * 在二分查找中判断订单是否是修改订单来对二分区间压缩
 * 判断订单是否是修改订单使用 “差分算法 ” 进行加速
 * 差分建议去看一下（实际上就是前缀和的逆运算）
 * 这里简略的说一下
 * 差分就是（这里是一维的）：有一个数组arr
 * 差分数组为 Differential[i] = arr[i] - arr[i-1]
 * 之后原数组的 arr[i] = sum(Differential[0] ~ Differential[i])(伪代码)
 * */
using namespace std;
int n, m;
const int N = 1000010;
int lessClassRoom[N]; // 每天剩余的教师数量
int start[N];//订单 i 借教室的起始日期
int endDay[N];// 订单 i 借教室的结束日期
int rooms[N];// 订单 i 借教师的数量
long long Differential[N];
class Solution {
public:
    // 二分判断条件函数
    // 使用差分算法进行加速
    bool check(int mid) {
        // 本处是用差分计算 ，该天需要的教室数量
        // 这后面一直说的原数组就是该天需要的教室数量
        memset(Differential, 0, sizeof Differential);//差分数组
        for (int i = 1; i <= mid; ++i) {
            // 差分数组 Differential[i] 加上或者减去 都会影响 i 之后原数组的值
            // 这里的这个订单从 start[i] 这天开始借 rooms[i] 间教室
            // 故利用差分数组的性质，start[i]~endDay[i]这几天都会借用 rooms[i]间教室
            // 故在 Differential[start[i]] 加上 rooms[i]
            // 即可得到下面的语句
            Differential[start[i]] += rooms[i];
            // 因为 endDay[i] + 1 开始这个订单就不用借教室了
            // 但是上边 Differential[start[i]] += rooms[i];
            // 会导致原数组[start[i]]后边所有加 rooms[i]
            // 但是endDay[i] + 1 开始这个订单就不用借教室了
            // 所以说 前边加上了 后边的减去 就等于没有改变
            Differential[endDay[i] + 1] -= rooms[i];
        }
        // 所谓的原数组 ，每天应该提供教室才能满足订单
        // 还原原数组并且查看哪天出问题，超过当天剩余的教室
        for (int i = 1; i <= n; ++i) {
            // 还原原数组
            Differential[i] = Differential[i- 1] + Differential[i];
            if (Differential[i] > lessClassRoom[i]) {
                return false;
            }
        }
        return true;
    }

    int brorrowClassRoom() {
        int left = 0, right = m;
        while (left < right) {
            int mid = (left + right + 1)>>1;
            if (check(mid)) {
                left = mid;
            } else {
                right = mid - 1;
            }
        }
        return right;
    }
};

int main() {
    Solution solution;
    cin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &lessClassRoom[i]);
    }
    for (int i = 1; i <= m; ++i) {
        scanf("%d%d%d", &rooms[i], &start[i], &endDay[i]);
    }
    int room = solution.brorrowClassRoom();
    if (room == m) {
        cout <<0;
        return 0;
    } else {
        cout << -1 << endl << room + 1;
    }
    return 0;
}