POJ 3667 Hotel（成段更新，區間合并）

題意：有N個房間，M次操作。1 a表示找到連續的長度為a的空房間，如果有多解，優先左邊的，即表示入住。2 b len把起點為b長度的len的房間清空，即退房。

用co作為延遲標記，如果為-1表示當前區間是兩個狀態，如果是0表示一種狀態，即區間內都沒有人入住，如果是1也表示一種狀態，表示滿客。線上段樹節點裡儲存三個值，端點左邊的(lmx)、右邊的(rmx)最長的連續的閒置房間數，總的閒置房間數目（mx）。查詢的時候，依次查詢端點左邊的連續的空閑房間，左子區間的，左子區間和右子區間串連起來，右區間部分是否滿足題意。區間合并還是一樣。如果左子區間的lmax等於左子區間的dis，那麼父區間的lmx，除了繼承左子區間lmx之外，還要加上右子區間的lmx。右端點的同理。

/*代碼風格更新後*/#include <iostream>#include <cstdio>#include <cstring>using namespace std;#define LL(x) (x<<1)#define RR(x) (x<<1|1)#define MID(a,b) (a+((b-a)>>1))const int N=50005;struct node{int lft,rht;int flag,mx,lmx,rmx;int len(){return rht-lft+1;}int mid(){return MID(lft,rht);}void init(){mx=lmx=rmx=len();}void fun(int tmp){if(tmp==1) mx=lmx=rmx=0;else mx=lmx=rmx=len();flag=tmp;//忘記添加延遲標記}};int n,m;struct Segtree{node tree[N*4];void down(int ind){if(tree[ind].flag){tree[LL(ind)].fun(tree[ind].flag);tree[RR(ind)].fun(tree[ind].flag);tree[ind].flag=0;}}void up(int ind){tree[ind].lmx=tree[LL(ind)].lmx;tree[ind].rmx=tree[RR(ind)].rmx;tree[ind].mx=max(tree[LL(ind)].mx,tree[RR(ind)].mx);if(tree[LL(ind)].lmx==tree[LL(ind)].len())tree[ind].lmx+=tree[RR(ind)].lmx;if(tree[RR(ind)].rmx==tree[RR(ind)].len())tree[ind].rmx+=tree[LL(ind)].rmx;tree[ind].mx=max(tree[ind].mx,max(tree[ind].lmx,tree[ind].rmx));tree[ind].mx=max(tree[ind].mx,tree[LL(ind)].rmx+tree[RR(ind)].lmx);}void build(int lft,int rht,int ind){tree[ind].lft=lft;tree[ind].rht=rht;tree[ind].flag=0;tree[ind].init();if(lft!=rht){int mid=tree[ind].mid();build(lft,mid,LL(ind));build(mid+1,rht,RR(ind));}}void updata(int st,int ed,int ind,int valu){int lft=tree[ind].lft,rht=tree[ind].rht;if(st<=lft&&rht<=ed) tree[ind].fun(valu);else{down(ind);int mid=tree[ind].mid();if(st<=mid) updata(st,ed,LL(ind),valu);if(ed> mid) updata(st,ed,RR(ind),valu);up(ind);}}int query(int valu,int ind){if(tree[ind].lft==tree[ind].rht) return tree[ind].lft;else{    down(ind);//查詢的時候,記得....            int pos;            if(tree[LL(ind)].mx>=valu) pos=query(valu,LL(ind));            else if(tree[LL(ind)].rmx+tree[RR(ind)].lmx>=valu) pos=tree[LL(ind)].rht-tree[LL(ind)].rmx+1;            else pos=query(valu,RR(ind));            up(ind);//查詢的時候,記得....            return pos;}}}seg;int main(){while(scanf("%d%d",&n,&m)!=EOF){seg.build(1,n,1);while(m--){int a,b,c,pos=0;scanf("%d",&a);if(a==1){scanf("%d",&b);if(seg.tree[1].mx>=b){pos=seg.query(b,1);seg.updata(pos,pos+b-1,1,1);}printf("%d\n",pos);}else{scanf("%d%d",&b,&c);seg.updata(b,b+c-1,1,-1);}}}return 0;}

/*代碼風格更新前*/#include <iostream>#include <cstdio>using namespace std;const int N=50005;struct node{    int left,right,co;    int lmax,rmax,mmax;    int mid(){return left+(right-left)/2;}    int dis(){return right-left+1;}    void change(int a)    {        co=a;        if(co==0) lmax=rmax=mmax=dis();        else lmax=rmax=mmax=0;    }};void unin(node &a,node &b,node &c){    a.lmax=b.lmax;  a.rmax=c.rmax;    a.mmax=max(max(b.mmax,c.mmax),b.rmax+c.lmax);    if(b.lmax==b.dis()) a.lmax+=c.lmax;    if(c.rmax==c.dis()) a.rmax+=b.rmax;}struct Segtree{    node tree[N*4];    void build(int left,int right,int r)    {        tree[r].left=left;  tree[r].right=right;        tree[r].lmax=tree[r].rmax=tree[r].mmax=tree[r].dis();        tree[r].co=-1;        if(left<right)        {            int mid=tree[r].mid();            build(left,mid,r*2);            build(mid+1,right,r*2+1);        }    }    void updata(int be,int end,int r,int co)    {        if(be<=tree[r].left&&tree[r].right<=end)        {            tree[r].change(co);        }        else        {            if(tree[r].co!=-1)            {                tree[r*2].change(tree[r].co);                tree[r*2+1].change(tree[r].co);                tree[r].co=-1;            }            int mid=tree[r].mid();            if(be<=mid) updata(be,end,r*2,co);            if(end>mid) updata(be,end,r*2+1,co);            unin(tree[r],tree[r*2],tree[r*2+1]);        }    }    int query(int len,int r)    {        if(tree[r].lmax>=len) return tree[r].left;        else if(tree[r*2].mmax>=len) return query(len,r*2);        else if(tree[r*2].rmax+tree[r*2+1].lmax>=len) return tree[r*2].right-tree[r*2].rmax+1;        else return query(len,r*2+1);    }}seg;int main(){    int n,m;    scanf("%d%d",&n,&m);    seg.build(1,n,1);    for(int i=0;i<m;i++)    {        int a,b,c;        scanf("%d",&a);        if(a==1)        {            scanf("%d",&b);            if(seg.tree[1].mmax>=b)            {                int pos=seg.query(b,1);                printf("%d\n",pos);                seg.updata(pos,pos+b-1,1,1);            }            else puts("0");        }        else        {            scanf("%d%d",&b,&c);            seg.updata(b,b+c-1,1,0);        }    }    return 0;}