Vijos P1782 借教室 ( 前缀和&&差分序列）

题目链接：借教室

 

题意：给出n天得教室数目，m个借教室得单子，按顺序借教室，问哪个单子不满足并输出

 

分析：可以用线段树做，会T，常数比较大，选择用差分序列维护前缀和，二分答案即可

 

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int ans,n,m,r[1001000],d[1001000],s[1001000],t[1001000],pre[1001000];
//inline int read(){int x=0;char ch=getchar();while(!(ch>='0'&&ch<='9')) ch=getchar();while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();};return x;}

int check(int x)
{
    int sum=0;
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=x;i++) pre[s[i]]+=d[i],pre[t[i]+1]-=d[i];
    for(int i=1;i<=n;i++)
    {
        sum+=pre[i];
        if(sum>r[i]) return 0;
    }    
    return 1;
}

int main()
{
    scanf("%d %d",&n,&m);
    ans=0;
    for(int i=1;i<=n;++i) scanf("%d",r+i);
    for(int i=1;i<=m;++i) {scanf("%d %d %d",d+i,s+i,t+i);}
    int l=1,r=m;
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(check(mid)==0) ans=mid,r=mid;
        else l=mid+1;
    }
    if(ans) printf("-1\n%d\n",ans);
    else puts("0"); 
}