AcWing 503. 借教室(二分/前缀和)

题目链接

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

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

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

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

我们需要处理接下来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≤n,m≤106,
0≤ri,dj≤109,
1≤sj≤tj≤n

输入样例：

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

输出样例：

-1 
2

思路：

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int N = 1e6 + 10;
const int M = 211;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
using namespace std;

int n,m;
int dp[N];
int d[N],s[N],t[N];
ll b[N];

bool judge(int k){
    rep(i,1,n) b[i]=dp[i];
    rep(i,1,k){
        b[s[i]]-=d[i];
        b[t[i]+1]+=d[i];
    }
    ll res=0;
    rep(i,1,n){
        res+=b[i];
        if(res<0) return 1;
    }
    return 0;
}

void solve(){
    cin>>n>>m;
    rep(i,1,n) cin>>dp[i];
    bep(i,n,1) dp[i]-=dp[i-1];
    rep(i,1,m) cin>>d[i]>>s[i]>>t[i];
    int l=1;
    int r=m;
    while(l<r){
        int mid=(l+r)>>1;
        if(judge(mid)) r=mid;
        else l=mid+1;
    }
    if(judge(r)){
        puts("-1");
        cout<<r<<endl;
    }
    else puts("0");
}

int main() {
    solve();
	return 0;
}