P1083 [NOIP2012 提高组] 借教室-二分+前缀和差分

[NOIP2012 提高组] 借教室

题目描述

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

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

我们需要处理接下来 $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

样例输入 #1

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

样例输出 #1

-1 
2

#include <bits/stdc++.h>
using namespace std;
#define de(x) cout<<x<<" ";
#define Pu puts("");
#define sf(x) scanf("%lld",&x);
typedef long long ll;//注意新增hack数据，需要定义long long
const int N=1e6+10,mod=100003;
ll pre[N];
ll a[N];
int n,m,ans;
ll d[N],zuo[N],you[N];
bool fun(int tmp_m){
    int u,x,y;
    int f=1;
    //超时做法：直接从第一个往后判断，直到找到第一个不满足的天数
    //通过看题解可以知道，对于第x天不满足，则它之后的每一天都不满足
    //满足单调性，所以考虑使用二分
    //for(int i=1;i<=m;i++)
    for(int i=1;i<=tmp_m;i++)
    {
        x=zuo[i];y=you[i];u=d[i];
        pre[x]-=u;//只需要修改左，右+1两个点，最后合并即可
        pre[y+1]+=u;
    }
    for(int j=1;j<=n+1;j++){//注意这里for循环的终点是n+1
        pre[j]+=pre[j-1];//合并
        //de(666)de(j)de(pre[j])de(a[j]+pre[j])de(666)Pu
        if(f==1&&a[j]+pre[j]<0) f=0;
        pre[j-1]=0;//把pre数组给恢复
    }
    return f;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        sf(a[i])
    }
    for(int i=1;i<=m;i++){
        sf(d[i])sf(zuo[i])sf(you[i])
    }
    int l=1,r=n+1,mid;
    if(fun(n)) printf("0\n");
    else{
        while(l<r){
        mid=(l+r)/2;
        if(fun(mid)) l=mid+1;//到第mid天满足
        else r=mid;
        }
        printf("-1\n%d\n",l);
    }
    return 0;
}