poj 3667 Hotel 线段树区间合并

至此,扫描线,线段树区间修改更新,可持久化线段树,线段树区间合并 等线段树题目全部过了一遍。

接下来还有好多高级数据结构等着我去学。

trick:线段树区间合并时,cover只需一个,因为一段区间只能处于一种状态,我作死开了两个cover分别对应两种情况,

好处是一目了然,坏处是更新完一个cover还要更新另一个,(因为一段区间只能处于一种状态) QAQ


#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <vector>
#define mid ((l+r)/2)
#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r
#define mk make_pair
using namespace std;
const int maxn=50000+7;
int siz[maxn*4];
int lsiz[maxn*4],rsiz[maxn*4];
int cover[maxn*4];  //out
int cover2[maxn*4];  //in
void build(int k,int l,int r){
    siz[k]=lsiz[k]=rsiz[k]=r-l+1;
    cover[k]=0;
    cover2[k]=0;
    if(l==r)return ;
    build(lson);
    build(rson);
}
void push_up(int k,int l,int r){
    siz[k]=max(siz[k<<1],max(siz[k<<1|1],rsiz[k<<1]+lsiz[k<<1|1]));
    if(lsiz[k<<1]==mid-l+1){
        lsiz[k]=lsiz[k<<1]+lsiz[k<<1|1];
    }
    else{
        lsiz[k]=lsiz[k<<1];
    }
    if(rsiz[k<<1|1]==r-mid){
        rsiz[k]=rsiz[k<<1|1]+rsiz[k<<1];
    }
    else {
        rsiz[k]=rsiz[k<<1|1];
    }
}
void push_down(int k,int l,int r){
    if(cover[k]){
        siz[k<<1]=lsiz[k<<1]=rsiz[k<<1]=mid-l+1;
        siz[k<<1|1]=lsiz[k<<1|1]=rsiz[k<<1|1]=r-mid;
        cover[k<<1]=cover[k<<1|1]=1;
        cover[k]=0;
        cover2[k<<1]=cover2[k<<1|1]=0;
    }
    if(cover2[k]){
    	siz[k<<1]=lsiz[k<<1]=rsiz[k<<1]=0;
        siz[k<<1|1]=lsiz[k<<1|1]=rsiz[k<<1|1]=0;
        cover2[k<<1]=cover2[k<<1|1]=1;
        cover2[k]=0;
        cover[k<<1]=cover[k<<1|1]=0;
    }
}
void update(int k,int l,int r,int a,int b,int v){
    if(a>r||b<l)return ;
    if(a<=l&&r<=b){
    	if(v==1)
    	{siz[k]=lsiz[k]=rsiz[k]=r-l+1;cover[k]=1;cover2[k]=0;}
    	if(v==2)
    	{siz[k]=lsiz[k]=rsiz[k]=0;cover2[k]=1;cover[k]=0;}
    	return ;
    }
    push_down(k,l,r);
    update(lson,a,b,v);
    update(rson,a,b,v);
    push_up(k,l,r);
}
int query(int k,int l,int r,int len){
	if(siz[k]<len)return 0;
	if(l==r)return l;
	if(lsiz[k]>=len)return l;
	push_down(k,l,r);
	if(siz[k<<1]>=len){
		return query(lson,len);
	}
	else if(rsiz[k<<1]+lsiz[k<<1|1]>=len){
		return mid-rsiz[k<<1]+1;
	}
	else return query(rson,len);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        build(1,1,n);
        for(int i=1;i<=m;i++){
            int t;
            scanf("%d",&t);
            if(t==1){
            	int num;scanf("%d",&num);
                int tmp=query(1,1,n,num);
                printf("%d\n",tmp);
                if(tmp)
                update(1,1,n,tmp,tmp+num-1,2);
            }
            else{
            	int l,d;scanf("%d%d",&l,&d);
                update(1,1,n,l,l+d-1,1);
            }
        }
    }
    return 0;
}















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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值