【NOIp 2012】【线段树】借教室

描述

在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。我们需要处理接下来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,第二行输出需要修改订单的申请人编号。

代码

线段树模板套一套

#include 
#include 
#include 
#define L(a) (a<<1)
#define R(a) (a<<1|1)
#define REP(a,b) for(int a=1;a<=b;a++)
#define maxn 1000030
using namespace std;

struct node{
    int lson,rson,val,lazy;
}t[maxn*3];

int n,m,r[maxn],ans=0;
bool flag=1;

int read(){
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        x=x*10+ch-'0'; ch=getchar();
    }
    return x*f;
}

void build_tree(int p,int l,int r){
    t[p].lson=l; t[p].rson=r; t[p].lazy=0;
    if(l==r) {
        t[p].val=read(); return;
    } else{
        int mid=(l+r)>>1;
        build_tree(L(p),l,mid);
        build_tree(R(p),mid+1,r);
        t[p].val=min(t[L(p)].val,t[R(p)].val);
        return;
    }
}

void pushdown(int p){
    if(t[p].lson==t[p].rson||t[p].lazy==0) return;
    t[L(p)].lazy+=t[p].lazy; t[R(p)].lazy+=t[p].lazy;
    t[L(p)].val-=t[p].lazy; t[R(p)].val-=t[p].lazy;
    t[p].lazy=0;
    t[p].val=min(t[L(p)].val,t[R(p)].val);
    return;
}

void modify(int p,int l,int r,int d){
    pushdown(p);
    if(t[p].lson==l&&t[p].rson==r){
        t[p].lazy+=d;
        t[p].val-=d;
        if(t[p].val<0) flag=0;
        return;
    } else {
        int mid=(t[p].lson+t[p].rson)>>1;
        if(r<=mid) modify(L(p),l,r,d);
        else if(l>mid) modify(R(p),l,r,d);
        else{
            modify(L(p),l,mid,d);
            modify(R(p),mid+1,r,d);
        }
        t[p].val=min(t[L(p)].val,t[R(p)].val);
        //if(t[p].val<0) flag=0;
        return;
    }
}

int query(int p,int l,int r){
    if(t[p].lson==l&&t[p].rson==r) return t[p].val;
    else {
        int mid=(l+r)>>1;
        if(r<=mid) return query(L(p),l,r);
        else if(l>mid) return query(R(p),l,r);
        else return min(query(L(p),l,mid),query(R(p),mid+1,r));
    }
}

int main(){
    scanf("%d%d",&n,&m);
    build_tree(1,1,n);
    while((++ans)<=m){
        int d=read(),s=read(),T=read();
        modify(1,s,T,d);
        if(!flag)  break;
    }
    if(flag) puts("0");
    else printf("-1\n%d\n",ans);
    return 0;
} 

你可能感兴趣的:(NOIp_刷题,NOIp_数据结构,线段树)