(poj 3667 Hotel)<丧病线段树区间合并>

传送门


Solution

第二道线段树区间合并的题
第一道稍微简单一点的线段树区间合并题戳这
细节看上一篇吧
一下是需要注意的地方
(来自poj讨论区)
假设max_len表示最大可用连续区间,start_left_len是左起最大可用连续区间,end_right_len是结束于right的最大可用连续区间,预定房间数为bookNum。则有:
1、如果1~n区间的max_len值都比bookNum小,就无解,否则有解。

2、如果start_left_len>=bookNum,返回left

3、如果它的左儿子的max_len值大于等于bookNum,到左儿子里去找。

4、如果左儿子的end_right_len加上右儿子的start_left_len大于等于bookNum,直接返回mid-leftChild.end_right_len+1,一定记得是直接返回,不需要再递归查询,我在这里WA了

5、否则到右儿子里去找。

上面每一步只有前面的每一步都不成立时,才能执行到下一步


Code

update和cl两个函数可以写成一个

// by spli
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;

const int N=50010;
int n,m;
struct node{
    int L,R,len;
    int ms,ls,rs;
    int tag;
}t[N*4+5];
int loc;

void pushup(int pos){
    t[pos].ls=t[pos<<1].ls;
    if(t[pos<<1].ms==t[pos<<1].len) t[pos].ls=max(t[pos].ls,t[pos<<1].ms+t[pos<<1|1].ls);
    t[pos].rs=t[pos<<1|1].rs;
    if(t[pos<<1|1].ms==t[pos<<1|1].len) t[pos].rs=max(t[pos].rs,t[pos<<1|1].ms+t[pos<<1].rs);
    t[pos].ms=max(t[pos<<1].ms,max(t[pos<<1|1].ms,t[pos<<1].rs+t[pos<<1|1].ls));
}

void build(int pos,int L,int R){
    t[pos].L=L;
    t[pos].R=R;
    t[pos].len=R-L+1;
    t[pos].tag=-1;
    if(L==R){
        t[pos].ls=1;
        t[pos].ms=1;
        t[pos].rs=1;
        return;
    }
    int mid=(L+R)>>1;
    build(pos<<1,L,mid);
    build(pos<<1|1,mid+1,R);
    pushup(pos);
}

void pushdown(int pos){
    if(t[pos].tag!=-1){
        t[pos<<1].ls=t[pos<<1].rs=t[pos<<1].ms=t[pos<<1].len*t[pos].tag;
        t[pos<<1|1].ls=t[pos<<1|1].rs=t[pos<<1|1].ms=t[pos<<1|1].len*t[pos].tag;
        t[pos<<1].tag=t[pos<<1|1].tag=t[pos].tag;
        t[pos].tag=-1;
    }
}

void update(int pos,int ll,int rr){
    int L=t[pos].L,R=t[pos].R;
    if(ll>R||L>rr) return;
    if(ll<=L&&rr>=R){
        t[pos].ls=t[pos].len;
        t[pos].rs=t[pos].len;
        t[pos].ms=t[pos].len;
        t[pos].tag=1;
        return;
    }
    pushdown(pos);
    int mid=(L+R)>>1;
    update(pos<<1,ll,rr);
    update(pos<<1|1,ll,rr);
    pushup(pos);
}

void cl(int pos,int ll,int rr){
    int L=t[pos].L,R=t[pos].R;
    if(ll>R||L>rr) return;
    if(ll<=L&&rr>=R){
        t[pos].ls=t[pos].rs=0;
        t[pos].ms=0;
        t[pos].tag=0;
        return;
    }
    pushdown(pos);
    cl(pos<<1,ll,rr);
    cl(pos<<1|1,ll,rr);
    pushup(pos);
}

void query(int pos,int k){
    pushdown(pos);
    if(t[pos].len<k||t[pos].ms<k) return;
    if(t[pos].ls>=k){
        loc=t[pos].L;
        cl(1,t[pos].L,t[pos].L+k-1);
        return;
    }
    else if(t[pos<<1].ms>=k) query(pos<<1,k);
        else if(t[pos<<1].rs+t[pos<<1|1].ls>=k){
            loc=t[pos<<1].R-t[pos<<1].rs+1;
            cl(1,loc,loc+k-1);
            return;
        }
    else query(pos<<1|1,k);
}

int main(){
    scanf("%d%d",&n,&m);
    build(1,1,n);
    int op,x,y;
    for(int i=1;i<=m;++i){
        scanf("%d",&op);
        if(op==1){
            scanf("%d",&x);
            if(t[1].ms<x){
                printf("0\n");
            }
            else {
                query(1,x);
                printf("%d\n",loc);
            }
        }
        else{
            scanf("%d%d",&x,&y);
            update(1,x,x+y-1);
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值