CSP-S 例题18-1 借教室

描述

在大学期间，经常需要租借教室。大到院系举办活动，小到学习小组自习讨论，都需要向学校申请借教室。教室的大小功能不同，借教室人的身份不同，借教室的手续也不一样。

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

我们需要处理接下来 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​，表示租借的数量，租借开始、结束分别在第几天。

每行相邻的两个数之间均用一个空格隔开。天数与订单均用从 1 开始的整数编号。

输出描述

如果所有订单均可满足，则输出只有一行，包含一个整数 0。否则（订单无法完全满足）

输出两行，第一行输出一个负整数 −1，第二行输出需要修改订单的申请人编号。

样例输入 1 

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

样例输出 1 

-1 
2

提示

数据范围与提示

对于 10% 的数据，有1≤n,m≤10；
对于 30% 的数据，有1≤n,m≤1000；
对于 70% 的数据，有1≤n,m≤10^5；
对于 100% 的数据，有1≤n,m≤10^6,0≤ri​,dj​≤10^9,1≤sj​≤tj​≤n。
【输入输出样例说明】
第 1 份订单满足后，4 天剩余的教室数分别为 0,3,2,3。第 2 份订单要求第 2 天到第 4 天每天提供 3 个教室，而第 3 天剩余的教室数为 2，因此无法满足。分配停止，通知第2 个申请人修改订单。

#include <bits/stdc++.h>
#define N 1000010
using namespace std;
typedef long long ll;
int n, m, a[N], b[N], c[N], d[N];
ll p[N];
bool classroom(int x)
{
    for (int i = 1; i <= n; i++)
    {
        p[i] = a[i] - a[i - 1];
    }
    for (int i = 1; i <= x; i++)
    {
        p[c[i]] -= b[i];
        p[d[i] + 1] += b[i];
    }
    ll cnt = 0;
    for (int i = 1; i <= n; i++)
    {
        cnt += p[i];
        if (cnt < 0)
        {
            return 1;
        }
    }
    return 0;
}
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    for (int i = 1; i <= m; i++)
    {
        cin >> b[i] >> c[i] >> d[i];
    }
    int temp = 1, ans = m;
    while (temp < ans)
    {
        int mid = (temp + ans) / 2;
        if (classroom(mid))
        {
            ans = mid;
        }
        else
        {
            temp = mid + 1;
        }
    }
    if (classroom(ans))
    {
        cout << -1 << endl
             << ans << endl;
    }
    else
    {
        cout << 0;
    }

    return 0;
}
/*
input:
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
output:
-1
2
*/