503. 借教室(acwing)

503. 借教室

题目描述

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

数据范围
1≤n,m≤10⁶,
0≤r_i,d_j≤10⁹,
1≤s_j≤t_j≤n
输入样例：

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

输出样例：

-1
2

二分查找+差分

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int z=1e6+10;  // 设定一个大数组的范围，用于存储数据

ll a[z],b[z];        // a数组用来存放每天可租借教室的数量，b是差分数组
ll c[z],c_l[z],c_r[z]; // c数组存放订单的教室数量，c_l和c_r分别存放订单的开始和结束天
ll copy_a[z];        // copy_a数组用于在check函数中复制a数组，以便进行计算
int n,m,i;           // n表示天数，m表示订单数量，i用于循环

// check函数检查是否能满足前mid个订单
bool check(ll mid)
{
    for(i=1;i<=n;i++)
    {
        b[i]=a[i]-a[i-1]; // 初始化差分数组b
        copy_a[i]=a[i];   // 复制a数组到copy_a，用于计算是否能满足订单
    }
    for(i=1;i<=mid;i++)
    {
        b[c_l[i]]-=c[i];  // 对于每个订单，减去开始租借的教室数量
        b[c_r[i]+1]+=c[i]; // 在结束租借的下一天，加回教室数量
    }
    for(i=1;i<=n;i++)
    {
        copy_a[i]=copy_a[i-1]+b[i]; // 从差分数组还原出实际每天可用的教室数量
        if(copy_a[i]<0) return true; // 如果某天教室数量不足，返回true
    }
    return false; // 所有订单都能满足，返回false
}

int main()
{
    scanf("%d %d",&n,&m); // 读取天数n和订单数m
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]); // 读取每天可租借的教室数量
    for(i=1;i<=m;i++)
        scanf("%lld %lld %lld",&c[i],&c_l[i],&c_r[i]); // 读取每个订单的详情

    // 检查是否所有订单均能满足
    if(!check(m))
    {
        printf("0"); // 如果能满足，输出0
        return 0;
    }
    
    // 二分查找无法满足的订单位置，左查找，查找左边界
    ll l=1,r=m; // 设置检查的范围
    ll mid;     // 用于存储中间值
    while(l<r)
    {
        mid=(l+r)>>1; // 找到中间值
        if(check(mid)) r=mid; // 如果前mid个订单无法满足，减小右边界
        else l=mid+1; // 如果能满足，增大左边界
    }
    printf("-1\n%d",r); // 输出-1和无法满足的订单编号
    return 0;
}

在这段代码中，我们首先使用差分数组来处理订单的教室租借问题。差分数组是一种处理数据变化的技巧，特别适合处理区间增减操作。通过对差分数组进行前缀和操作，我们可以还原出每天实际可用的教室数量，并检查是否所有的订单都可以被满足。

如果不能满足所有订单，我们使用二分查找来确定第一个无法满足的订单。这是通过不断缩小检查的订单范围实现的，直到找到一个点，在这一点上，如果包括这个订单，就无法满足所有订单的需求。这样做的目的是找到那个特定的订单，我们需要告知申请者修改他们的订单。