P1083 [NOIP2012 提高组] 借教室

[NOIP2012 提高组] 借教室 - 洛谷

题目描述

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

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

我们需要处理接下来 nn 天的借教室信息，其中第 ii 天学校有 r_iri​ 个教室可供租借。共有 mm 份订单，每份订单用三个正整数描述，分别为 d_j,s_j,t_jdj​,sj​,tj​，表示某租借者需要从第 s_jsj​ 天到第 t_jtj​ 天租借教室（包括第 s_jsj​ 天和第 t_jtj​ 天），每天需要租借 d_jdj​ 个教室。

我们假定，租借者对教室的大小、地点没有要求。即对于每份订单，我们只需要每天提供 d_jdj​ 个教室，而它们具体是哪些教室，每天是否是相同的教室则不用考虑。

借教室的原则是先到先得，也就是说我们要按照订单的先后顺序依次为每份订单分配教室。如果在分配的过程中遇到一份订单无法完全满足，则需要停止教室的分配，通知当前申请人修改订单。这里的无法满足指从第 s_jsj​ 天到第 t_jtj​ 天中有至少一天剩余的教室数量不足 d_jdj​ 个。

现在我们需要知道，是否会有订单无法完全满足。如果有，需要通知哪一个申请人修改订单。

输入格式

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

第二行包含 nn 个正整数，其中第 ii 个数为 r_iri​，表示第 ii 天可用于租借的教室数量。

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

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

输出格式

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

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

二分答案即可，如果直接模拟的话复杂度是O（mn），二分之后是O（nlogm）。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int a[N];
ll cf[N];
int l[N],r[N];
int s[N];
int n,m;
bool check(int id)
{
    memset(cf,0,sizeof(cf));
    for(int i=1;i<=id;i++)
    {
        cf[l[i]]-=s[i];
        cf[r[i]+1]+=s[i];
    }
    for(int i=1;i<=n;i++)cf[i]+=cf[i-1];
    for(int i=1;i<=n;i++)
    {
        if((ll)a[i]+cf[i]<0)return false;
    }
    return true;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=m;i++)cin>>s[i]>>l[i]>>r[i];
    if(!check(1)){
        cout<<-1<<endl<<1;
        //cout<<"houzi"<<endl;
        return 0;
    }
    int li=1,ri=m+1;
    while(ri-li>1)
    {
        int mid=(li+ri)>>1;
        if(check(mid))li=mid;
        else ri=mid;
    }
    if(li>=m){
        cout<<0<<endl;
        return 0;
    }
    cout<<-1<<endl<<li+1;
    return 0;
}