首页 > 代码库 > 【NOIP2012】借教室

【NOIP2012】借教室

在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要 向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。 
面对海量租借教室的信息,我们自然希望编程解决这个问题。 
我们需要处理接下来n天的借教室信息,其中第i天学校有ri个教室可供租借。共有m份 订单,每份订单用三个正整数描述,分别为dj,sj,tj,表示某租借者需要从第sj天到第tj天租 借教室(包括第sj天和第tj天),每天需要租借dj个教室。 
我们假定,租借者对教室的大小、地点没有要求。即对于每份订单,我们只需要每天提 供dj个教室,而它们具体是哪些教室,每天是否是相同的教室则不用考虑。 
借教室的原则是先到先得,也就是说我们要按照订单的先后顺序依次为每份订单分配教 室。如果在分配的过程中遇到一份订单无法完全满足,则需要停止教室的分配,通知当前申 请人修改订单。这里的无法满足指从第sj天到第tj天中有至少一天剩余的教室数量不足dj个。 
现在我们需要知道,是否会有订单无法完全满足。如果有,需要通知哪一个申请人修改 订单。

Input

第一行包含两个正整数n,m,表示天数和订单的数量。 
第二行包含n个正整数,其中第i个数为ri,表示第i天可用于租借的教室数量。
接下来有m行,每行包含三个正整数dj,sj,tj,表示租借的数量,租借开始、结束分别在 第几天。 
每行相邻的两个数之间均用一个空格隔开。天数与订单均用从1开始的整数编号。

Output

如果所有订单均可满足,则输出只有一行,包含一个整数 0。否则(订单无法完全满足) 输出两行,第一行输出一个负整数-1,第二行输出需要修改订单的申请人编号。

 
 
题解:
  oj的标签是平衡树,其实只要一颗线段树就可以了,你想我们需要一个nlogn的算法,显然是用数据结构去优化暴力,我们要维护的是每天剩余的教室数,但仔细分析一下,不难发现,因为每个询问都是对于一个区间,那么只要区间的最小值够询问才能满足,所以可以考虑用线段树维护一下区间的最小值,当然减的时候一定要打懒标记,不然复杂度就不对了。
代码:
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
const int MAXN=1000010;
int day[MAXN],n,m;
struct tree{
    int l,r,m,lz;
}a[MAXN*4];
 
int min(int x,int y){
    if(x<y) return x;
    return y;
}
 
void build(int id,int l,int r){
    if(l==r){
        a[id].l=l,a[id].r=r;
        a[id].m=day[l],a[id].lz=0;
        return;
    }
    a[id].l=l,a[id].r=r,a[id].lz=0;
    int mid=(l+r)/2;
    build(id*2,l,mid);
    build(id*2+1,mid+1,r);
    a[id].m=min(a[id*2].m,a[id*2+1].m);
}
 
void lazz(int id){
    if(a[id].lz!=0){
        a[id*2].lz+=a[id].lz,a[id*2+1].lz+=a[id].lz;
        a[id*2].m-=a[id].lz,a[id*2+1].m-=a[id].lz;
        a[id].lz=0;
        return;
    }
}
 
int kanxun(int l,int r,int id){
    int L=a[id].l,R=a[id].r,mid=(L+R)/2;
    if(L==l&&R==r){
        return a[id].m;
    }
    lazz(id);
    if(r<=mid) return kanxun(l,r,id*2);
    else if(l>mid) return kanxun(l,r,id*2+1);
    else return min(kanxun(l,mid,id*2),kanxun(mid+1,r,id*2+1));
    //这里之所以不要加update是因为我们认为,下放懒标记之后的已经把
    //区间所存的数修改正确,而不受儿子结点的影响,因为儿子的laze就是从爸爸那拿来的
}
 
void update(int l,int r,int id,int x){
    int L=a[id].l,R=a[id].r,mid=(L+R)/2;
    if(L==l&&R==r){
        a[id].lz+=x;
        a[id].m-=x;
        return;
    }
    lazz(id);
    if(r<=mid) update(l,r,id*2,x);
    else if(l>mid) update(l,r,id*2+1,x);
    else update(l,mid,id*2,x),update(mid+1,r,id*2+1,x);
    a[id].m=min(a[id*2].m,a[id*2+1].m);
}
 
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&day[i]);
    build(1,1,n);
    for(int i=1;i<=m;i++){
        int z,x,y;
        scanf("%d%d%d",&z,&x,&y);
        int have=kanxun(x,y,1);
        if(have<z){printf("-1\n%d",i);return 0;}
        update(x,y,1,z);
    }
    printf("0\n");
    return 0;
}

 

【NOIP2012】借教室