poj 3669 线段树成段更新+区间合并

添加 lsum[ ] , rsum[ ] , msum[ ] 来记录从左到右的区间，从右到左的区间和最大的区间；

#include<stdio.h>

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

#define maxn 50005

int rsum[maxn<<2],lsum[maxn<<2],msum[maxn<<2];//msum[]维护区间1…N中的最大连续区间长度

int mark[maxn<<2];

int max(int x,int y)

{

    return x>y?x:y;

}

void pushup(int l,int r,int rt)

{

    int m=(l+r)/2;

    lsum[rt]=lsum[rt<<1];

    rsum[rt]=rsum[rt<<1|1];

    msum[rt]=max(msum[rt<<1],msum[rt<<1|1]);

    if(lsum[rt<<1]==m-l+1)

        lsum[rt]=lsum[rt<<1]+lsum[rt<<1|1];

    if(rsum[rt<<1|1]==r-m)

        rsum[rt]=rsum[rt<<1|1]+rsum[rt<<1];

    msum[rt]=max(msum[rt],lsum[rt<<1|1]+rsum[rt<<1]);

}

void pushdown(int l,int r,int rt)

{

    if(mark[rt]!=-1)

    {

        int len=r-l+1;



        mark[rt<<1]=mark[rt];

        mark[rt<<1|1]=mark[rt];

        if(mark[rt]==0)

        {

            lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1]=0;

            lsum[rt<<1|1]=rsum[rt<<1|1]=msum[rt<<1|1]=0;

        }

        else

        {

            lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1]=len-len/2;

            rsum[rt<<1|1]=lsum[rt<<1|1]=msum[rt<<1|1]=len/2;

        }

        mark[rt]=-1;

    }

}

void build(int l,int r,int rt)

{

    mark[rt]=-1;

    if(l==r)

    {

        lsum[rt]=rsum[rt]=msum[rt]=1;

        return ;

    }

    int m=(l+r)/2;

    build(lson);

    build(rson);

    pushup(l,r,rt);

}

void updata(int L,int R,int c,int l,int r,int rt)

{

    if(l>=L&&R>=r)

    {

        mark[rt]=c;

        if(mark[rt]==0)

        {

            lsum[rt]=rsum[rt]=msum[rt]=0;

        }

        else 

            lsum[rt]=rsum[rt]=msum[rt]=r-l+1;

        return ;

    }

    pushdown(l,r,rt);

    int m=(l+r)/2;

    if(m>=L)

        updata(L,R,c,lson);

    if(R>m)

        updata(L,R,c,rson);

    pushup(l,r,rt);

}

int query(int num,int l,int r,int rt)

{

    if(l==r)

    {

        return l;

    }

    pushdown(l,r,rt);

    int m=(l+r)/2;

    int ret;

    if(msum[rt<<1]>=num)//由于从最左边开始坐位子，所以rsum[rt<<1]>

    {

        return query(num,lson);

    }

    else if(lsum[rt<<1|1]+rsum[rt<<1]>=num)

        return m-rsum[rt<<1]+1;

    else

        return query(num,rson);

}

int main()

{

    int n,m;

    while(scanf("%d%d",&n,&m)!=EOF)

    {

        build(1,n,1);

        int a,b,c;

        while(m--)

        {

            scanf("%d",&a);

            if(a==1)

            {

                scanf("%d",&b);

                if(b>msum[1])

                    printf("0\n");

                else

                {

                    int ret=query(b,1,n,1);

                    printf("%d\n",ret);

                    updata(ret,ret+b-1,0,1,n,1);

                }

            }

            else

            {

                scanf("%d%d",&b,&c);

                updata(b,b+c-1,1,1,n,1);

            }

        }

    }

}