Luogu P1083 借教室

Luogu P1083 借教室

上周五做的
感觉和会场预约挺像的。。不过会场预约我没做。。。
本来想着学学这道题线段树的解法，不过发现前缀和+二分的做法更简单
推荐这篇blog

偶然发现了师弟的题解 好优秀啊…
真是惭愧。。。

很神奇的一道题。
中午精神恍惚。。。唉浪费了不少时间。。

#include<cstdio>
#include<cstring>

int n,m,l,r;
int la[1000100],v[1000100],st[1000100],ed[1000100],day[1000100],a[1000100];

int check(int x)
{
    memset(v,0,sizeof(v));
    for(int i=1;i<=x;i++)
    {
        v[st[i]]+=day[i];
        v[ed[i]+1]-=day[i];
    }
    for(int i=1;i<=n;i++)
    {
        la[i]=la[i-1]+v[i];

        if(la[i]>a[i])
            return 0;
    }
    return 1;
}

int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)
        scanf("%d %d %d",&day[i],&st[i],&ed[i]);
    l=1;r=m;
    while(l<r)
    {
        int mid=(l+r)/2;
        if(check(mid))
        {
            l=mid+1;
        }
        else r=mid;
    }
    if(l==m)
        printf("0\n");
    else 
        printf("-1\n%d",l);
}

第一次交时因为数组小了还RE了。。。