蓝桥杯经典二分例题：借课室

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

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

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

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

我们需要处理接下来 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，第二行输出需要修改订单的申请人编号。

数据范围

1≤n,m≤1000000
0≤ri,dj≤1000000000

1≤sj，tj≤n

输入样例：

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

输出样例：

-1
2

第一个做法：暴力，时间复杂度n的三次方，只能过一部分样例

#include<bits/stdc++.h>
using namespace std;
int main()
{
    const int N=1e6+10;
    int a[N],b[N];
    memset(b,0,sizeof b);
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++) cin>>a[i];
    for(int j=1;j<=m;j++)
    {
        int d,s,t;
        cin>>d>>s>>t;
        for(int i=s;i<=t;i++) b[i]+=d;
        for(int i=1;i<=n;i++) 
        {
            if(b[i]>a[i])
            {
                cout<<"-1"<<endl<<j<<endl;
               return 0; 
            } 
        
        } 
        
    }
    cout<<"0"<<endl;
    return 0;
   
}

 

第二个做法：利用从某一个订单后的所有订单都无法满足的二段特性进行二分 ，并利用差分来计算第s天到第t天的房间预定数，可ac

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+100;
int n,m;
int w[N];
int d[N],s[N],t[N];
long long b[N];

bool check(int mid)
{
    memset(b,0,sizeof b);
    for(int i=1;i<=mid;i++)
    {
        b[s[i]]+=d[i];
        b[t[i]+1]-=d[i];//差分
    }
    long long s=0;//记录需要的教室数量
    for(int i=1;i<=n;i++)
    {
        s+=b[i];
        if(s>w[i]) return false;//教室不够用，返回false
    }
    return true;
    
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>w[i];
    for(int i=1;i<=m;i++) cin>>d[i]>>s[i]>>t[i];
    int l=0,r=m;
    while(l<r)//二分，利用第k订单时够用，第k+1单之后不够用的二段性，把订单二分
    {
        int mid=(l + r + 1)/2;
        if(check(mid)) l=mid;
        else r=mid-1;
    }
    if(r==m) cout<<"0"<<endl;
    else cout<<"-1"<<endl<<r+1; 
    return 0;
   
}