[NOIp2012提高组]借教室

OJ题号：洛谷1083

思路：ZKW线段树

#include<cstdio>
#include<cctype>
#include<algorithm>
inline int getint() {
    char ch;
    while(!isdigit(ch=getchar()));
    int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=1000001;
class ZKW {
    #define _left <<1
    #define _right <<1|1
    private:
        int m,val[N<<1],tag[N<<1];
    public:
        void build(const int n) {
            for(m=1;m<=n;m<<=1);
            for(int i=1;i<=n;i++) val[m+i]=getint();
            for(int i=m;i;i--) val[i]=std::min(val[i _left],val[i _right]);
        }
        void modify(int l,int r,int x) {
            for(l+=m-1,r+=m+1;l^r^1;l>>=1,r>>=1) {
                if(l<m) val[l]=std::min(val[l _left],val[l _right])-tag[l];
                if(~l&1) val[l^1]-=x,tag[l^1]+=x;
                if(r<m) val[r]=std::min(val[r _left],val[r _right])-tag[r];
                if(r&1) val[r^1]-=x,tag[r^1]+=x;
            }
            for(;l&&r;l>>=1,r>>=1) {
                val[l]=std::min(val[l _left],val[l _right])-tag[l];
                val[r]=std::min(val[r _left],val[r _right])-tag[r];
            }
        }
        bool check() {
            return val[1]>=0;
        }
};
ZKW tree;
int main() {
    int n=getint(),m=getint();
    tree.build(n);
    for(int i=1;i<=m;i++) {
        int d=getint(),s=getint(),t=getint();
        tree.modify(s,t,d);
        if(!tree.check()) {
            printf("-1
%d
",i);
            return 0;
        }
    }
    puts("0");
    return 0;
}