POJ3667—線段樹

//對連續空間的插入和刪除//snum---表示此節點下空間個數，ls---表示此節點從左開始的空間個數，rs---表示此節點從右開始空間個數//在查詢時，按左節點、左節點---右節點、右節點的次序查詢//在每次查詢時都要向下更新；在查詢成功時，要向上更新#include <iostream>#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;#define N 50005#define inf 123456789typedef __int64 i64;int n,m,t;struct TNode{    int l,r,snum;    int ls,rs;} node[6*N];void build(int root,int l,int r){    int mid=(l+r)>>1,p=root<<1;    node[root].l=l,node[root].r=r;    node[root].snum=node[root].ls=node[root].rs=r-l+1;    if(l==r) return ;    build(p,l,mid);    build(p+1,mid+1,r);}void update(int root)//向下更新{    int p=root<<1;    if(node[root].snum==0)    {        node[root].ls=node[root].rs=0;        node[p].snum=node[p+1].snum=0;        node[p].ls=node[p].rs=node[p+1].ls=node[p+1].rs=0;    }    if(node[root].snum==node[root].r-node[root].l+1)    {        node[root].ls=node[root].rs=node[root].r-node[root].l+1;        node[p].ls=node[p].rs=node[p].snum=node[p].r-node[p].l+1;        node[p+1].ls=node[p+1].rs=node[p+1].snum=node[p+1].r-node[p+1].l+1;    }}void up(int root)//向上更新{    int p=root<<1;    node[root].snum=node[p].snum+node[p+1].snum;    node[root].ls=node[p].ls;    if(node[p].ls==node[p].r-node[p].l+1)    {        node[root].ls+=node[p+1].ls;    }    node[root].rs=node[p+1].rs;    if(node[p+1].rs==node[p+1].r-node[p+1].l+1)    {        node[root].rs+=node[p].rs;    }}void modin(int root,int l,int r){    update(root);    if(node[root].l==l&&node[root].r==r)    {        node[root].snum=node[root].ls=node[root].rs=0;    }    else    {        int mid=(node[root].l+node[root].r)>>1,p=root<<1;        if(l<=mid) modin(p,l,min(r,mid));        if(r>mid) modin(p+1,max(mid+1,l),r);        up(root);    }}void modout(int root,int l,int r){    update(root);    if(node[root].l==l&&node[root].r==r)    {        node[root].snum=node[root].ls=node[root].rs=r-l+1;    }    else    {        int mid=(node[root].l+node[root].r)>>1,p=root<<1;        if(r<=mid) modout(p,l,r);        else if(l>mid) modout(p+1,l,r);        else        {            modout(p,l,mid);            modout(p+1,mid+1,r);        }        up(root);    }}bool query(int root,int len){    update(root);    if(len>node[root].snum) return 0;    if(node[root].snum==node[root].r-node[root].l+1||node[root].ls>=len)    {        printf("%d\n",node[root].l);        modin(root,node[root].l,node[root].l+len-1);        up(root);        return 1;    }    int p=root<<1;    if(query(p,len))    {        up(root);        return 1;    }    if(node[p].rs+node[p+1].ls>=len)    {        int prs=node[p].rs;        modin(p,node[p].r-prs+1,node[p].r);        modin(p+1,node[p+1].l,node[p+1].l+len-prs-1);        up(root);        printf("%d\n",node[p].r-prs+1);        return 1;    }    if(query(p+1,len))    {        up(root);        return 1;    }}int main(){    int id,l,r;    scanf("%d%d",&n,&m);    build(1,1,n);    while(m--)    {        scanf("%d",&id);        if(id==1)        {            scanf("%d",&l);            if(!query(1,l)) puts("0");        }        else        {            scanf("%d%d",&l,&r);            modout(1,l,l+r-1);        }    }    return 0;}