poj hotel线段树区间合并

节点主要记录三个信息:msum,lsum,rsum;

假设,L---R代表当前节点代表区间;

msum表示 在L————R区间内最大连续空房间数目;

lsum表示在L————R区间内以L为起点的最大连续空房间数目;

rsum表示在L————R区间内以R为右端点(自由向左的)的最大连续空房间数目;

#include<cstdio>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1 | 1
const int N=50002;
int msum[N<<2],rsum[N<<2],lsum[N<<2],col[N<<2];
void PushUp(int len,int rt){
    lsum[rt]=lsum[rt<<1];
    rsum[rt]=rsum[rt<<1 | 1];
    if(lsum[rt]==len-(len>>1)) lsum[rt] += lsum[rt<<1|1];
    if(rsum[rt]==len>>1) rsum[rt] += rsum[rt<<1];
    msum[rt]=max(rsum[rt<<1]+lsum[rt<<1|1],max(msum[rt<<1],msum[rt<<1 | 1]));
}
void PushDown(int len,int rt){
  if(col[rt]!=-1){
        col[rt<<1]=col[rt<<1|1]=col[rt];
        msum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=col[rt]?0:(len-(len>>1));
        msum[rt<<1 | 1]=lsum[rt<<1 | 1]=rsum[rt<<1 | 1]=col[rt]?0:(len>>1);
        col[rt]=-1;
   }
}
void build(int l,int r ,int rt){
    msum[rt]=lsum[rt]=rsum[rt]=r-l+1;
    col[rt]=-1;
    if(l==r)return;
     int mid=(l+r)>>1;
     build(lson);
      build(rson);
}
void update(int L,int R,int val,int l,int r,int rt){
        if(L<=l&&r<=R){
            msum[rt]=lsum[rt]=rsum[rt]=val?0:(r-l+1);
            col[rt]=val;
            return;
         }
         PushDown(r-l+1,rt);
        int mid=(l+r)>>1;
        if(L<=mid)
          update(L,R,val,lson);
        if(mid<R)
          update(L,R,val,rson);
        PushUp(r-l+1,rt);
}
int query(int val,int l,int r,int rt){
        if(l==r)
         return l;
        PushDown(r-l+1,rt);
        int mid=(l+r)>>1;
        if(msum[rt<<1] >= val)
        {
           return query(val,lson);
        }
        else if(lsum[rt<<1|1]+rsum[rt<<1]>=val){
             return mid-rsum[rt<<1]+1;
        }
        else
        return query(val,rson);

}
int main(){
  int n,m;
  while(~scanf("%d%d",&n,&m)){
    build(1,n,1);
    int a,b,cnt;
    while(m--){
          scanf("%d",&a);
            if(a==1){
             scanf("%d",&b);
              if(msum[1]<b){
               printf("0\n");
               }
              else{
                int id=query(b,1,n,1);
                 printf("%d\n",id);
                update(id,id+b-1,1,1,n,1);
              }
            }
        else if(a==2){
          scanf("%d %d",&b,&cnt);
          update(b,b+cnt-1,0,1,n,1);
        }
    }
  }
        return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值