[洛谷]P1083 借教室

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>


using namespace std;


#define maxn 1000005


struct node{
    int st,ed,room;
}clss[maxn];


int room_num[maxn],n,m,a[maxn];


inline bool find(int x)//二分方案
    {
        memset(a,0,sizeof(a));
        for(int i=1;i<=x;i++)
            {
                a[clss[i].st]+=clss[i].room; //a[i]表示第i天需要教室数
                a[clss[i].ed+1]-=clss[i].room;
            }
        for(int i=1;i<=n;i++)
            {
                a[i]+=a[i-1]; //当天需求
                if(a[i]>room_num[i]) return true;//需求超过可用教室数，说明方案不可行
            }
        return false;
    }


int main()
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",&room_num[i]);
        for(int i=1;i<=m;i++)
            scanf("%d%d%d",&clss[i].room,&clss[i].st,&clss[i].ed);
        int l=1,r=m;
        while(l<r)
            {
                int mid=(l+r)>>1;
                if(find(mid)) r=mid; //往前查找第一个不符合条件的方案
                else l=mid+1;//往后查找
            }
        if(m==r) printf("0\n"); //没有找到过不符合条件的方案
        else printf("-1\n%d\n",r);
        return 0;
    }