【线段树】POJ 3667 Hotel 区间合并

来源：互联网发布：淘宝怎么打单编辑：程序博客网时间：2024/04/28 17:58

水题好像来的晚了一些

题意:

1 a:询问是不是有连续长度为a的空房间,有的话住进最左边

2 a b:将[a,a+b-1]的房间清空

在最底层用1表示空，0表示占有这房间。。

sum[rt]表示该节点下最大连续空房间数

lsum[rt]表示该节点从左端开始连续的空房间数

rsum[rt]表示该节点从右端开始连续的空房间数

#include <stdio.h>#include <string.h>#include <stdlib.h>#include <math.h>#include <string>#include <iostream>#include <algorithm>using namespace std;#include <queue>#include <stack>#include <vector>#include <deque>#include <set>#include <map>#define IN     freopen ("in.txt" , "r" , stdin);#define OUT  freopen ("out.txt" , "w" , stdout);typedef long long  LL;const int MAXN = 100110;//点数的最大值const int MAXM = 20006;//边数的最大值const int INF = 11521204;const int mod=1000000007;#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1int sum[MAXN<<2],lsum[MAXN<<2],rsum[MAXN<<2];int col[MAXN<<2];void pushup(int rt,int m){    lsum[rt]=lsum[rt<<1];    rsum[rt]=rsum[rt<<1|1];    if(lsum[rt]==m-(m>>1))  lsum[rt]+=lsum[rt<<1|1];    if(rsum[rt]==m>>1)  rsum[rt]+=rsum[rt<<1];    sum[rt]=max(max(sum[rt<<1],sum[rt<<1|1]),rsum[rt<<1]+lsum[rt<<1|1]);}void pushdown(int rt,int m){    if(col[rt]!=-1)    {        col[rt<<1]=col[rt<<1|1]=col[rt];        sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=col[rt]?0:m-(m>>1);        sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=col[rt]?0:m>>1;        col[rt]=-1;    }}void build(int l,int r,int rt){    sum[rt]=lsum[rt]=rsum[rt]=r-l+1;    col[rt]=-1;    if(l==r)        return ;    int m=(l+r)>>1;    build(lson);    build(rson);}void update(int L,int R,int cc,int l,int r,int rt){    if(L<=l&&r<=R)    {        sum[rt]=lsum[rt]=rsum[rt]=cc?  0:r-l+1;        col[rt]=cc;        return ;    }    pushdown(rt,r-l+1);    int m=(l+r)>>1;    if(L<=m)    update(L,R,cc,lson);    if(m<R)    update(L,R,cc,rson);    pushup(rt,r-l+1);}int query(int x,int l,int r,int rt){    if(l==r)        return l;    pushdown(rt,r-l+1);    int m=(l+r)>>1;    if(sum[rt<<1]>=x)        return query(x,lson);    else if(rsum[rt<<1]+lsum[rt<<1|1]>=x)        return m-rsum[rt<<1]+1;    else return query(x,rson);}int main(){    int n,m;    //IN;    while(scanf("%d%d",&n,&m)!=EOF)    {        build(1,n,1);        while(m--)        {            int op,k,d;            scanf("%d",&op);            if(op==1)            {                scanf("%d",&d);                if(sum[1]<d)                    printf("0\n");                else                {                    int a=query(d,1,n,1);                    printf("%d\n",a);                    update(a,a+d-1,1,1,n,1);                }            }            else            {                scanf("%d%d",&k,&d);                update(k,k+d-1,0,1,n,1);            }        }    }    return 0;}

0 0