Hotel(线段树区间合并)

题目
有n个房间,m次操作,操作 1 x表示有x个人要入住,并且房间要挨着,就是要找到连续的x个房间,输出左端点值,操作 2 x ,y表示x~x+y-1的房间清空了。
又是日常不会然后去百度,线段树花样也太多了吧。
维护三个值sum,lmax,rmax,分别表示区间最大连续,从左端点开始的最大连续,和从右端点开始的最大连续。
t r e e [ n o d e ] . s u m 显 然 是 M a x ( t r e e [ n o d e < < 1 ] . s u m , t r e e [ n o d e < < 1 ∣ 1 ] . s u m , t r e e [ n o d e < < 1 ] . r m a x + t r e e [ n o d e < < 1 ∣ 1 ] . l m a x ) . tree[node].sum显然是Max(tree[node<<1].sum, tree[node<<1|1].sum,tree[node<<1].rmax+tree[node<<1|1].lmax). tree[node].sumMax(tree[node<<1].sum,tree[node<<11].sum,tree[node<<1].rmax+tree[node<<11].lmax).因为区间的最大连续要么在左区间要么再右区间,要么在左右区间合并后的中间部分。确定关系之后就可以开始写了。

代码

#include <iostream>
#include <cstring>
#include <cmath>
#include <bitset>
#include <queue>
#include <vector>
#include <cstdio>
#include <set>
#include <sstream>
#include <string>
#include <algorithm>
#include <map>
#define rep(i, a, b) for (int i = a; i <= b; i++)
#define frep(i, a, b) for (int i = a; i >= b; i--)
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/gcd(a,b)
using namespace std;
const int N = 1e7 + 7;
const int M = 1e2+7;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;
typedef long long ll;
int test_prime[8]={2,3,5,7,11,13,17,19};
struct node
{
    int sum,lmax,rmax,lazy;
}tree[N<<2];
void build(int node,int l,int r)
{
    tree[node].sum=tree[node].lmax=tree[node].rmax=r-l+1;
    tree[node].lazy=-1;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(node<<1,l,mid);
    build(node<<1|1,mid+1,r);
}
void pushup(int node,int l,int r)
{
    int mid=(l+r)>>1;
    tree[node].lmax=tree[node<<1].lmax;
    if(tree[node<<1].sum==mid-l+1)tree[node].lmax=tree[node<<1].sum+tree[node<<1|1].lmax;
    //左区间全部连续,所以可以和右区间的左区间合并
    tree[node].rmax=tree[node<<1|1].rmax;
    if(tree[node<<1|1].sum==r-mid)tree[node].rmax=tree[node<<1].rmax+tree[node<<1|1].sum;//同上
    tree[node].sum=max(tree[node<<1].rmax+tree[node<<1|1].lmax,max(tree[node<<1].sum,tree[node<<1|1].sum));
}
void pushdown(int node,int l,int r)
{
    int mid=(l+r)>>1;
    tree[node<<1].lazy=tree[node<<1|1].lazy=tree[node].lazy;
    tree[node<<1].sum=tree[node<<1].lmax=tree[node<<1].rmax=(mid-l+1)*tree[node].lazy;
    //lazy取0,1分别表示入住和清空
    tree[node<<1|1].sum=tree[node<<1|1].lmax=tree[node<<1|1].rmax=(r-mid)*tree[node].lazy;
    tree[node].lazy=-1;
}
void update(int node,int L,int R,int l,int r,int s)
{
    if(L<=l&&r<=R){
        tree[node].lazy=s;
        tree[node].sum=tree[node].lmax=tree[node].rmax=(r-l+1)*s;
        return;
    }
    if(tree[node].lazy!=-1){
        pushdown(node,l,r);
    }
    int mid=(l+r)>>1;
    if(L<=mid)update(node<<1,L,R,l,mid,s);
    if(R>mid)update(node<<1|1,L,R,mid+1,r,s);
    pushup(node,l,r);
}
int query(int node,int val,int l,int r)
{
    if(tree[node].lazy!=-1){
        pushdown(node,l,r);
    }
    if(l==r)return l;
    int mid=(l+r)>>1;
    if(tree[node<<1].sum>=val)return query(node<<1,val,l,mid);
    else if(tree[node<<1].rmax+tree[node<<1|1].lmax>=val){
        return mid-tree[node<<1].rmax+1;
        //rmax+lmax>=val,说明中间部分可以。因为mid是左区间的右端点,所以开始位置就是mid-rmax+1.
    }
    else return query(node<<1|1,val,mid+1,r);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    rep(i,1,m){
        int op;
        scanf("%d",&op);
        if(op&1){
            int v;
            scanf("%d",&v);
            if(v>tree[1].sum){
                puts("0");
                continue;
            }
            int pos=query(1,v,1,n);
            printf("%d\n",pos);
            update(1,pos,pos+v-1,1,n,0);
        }else{
            int u,v;
            scanf("%d%d",&u,&v);
            update(1,u,u+v-1,1,n,1);
        }
    }
    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值