PKU2892(Tunnel Warfare)线段树求最大连续区间

/**************************************
题目大意:
一条直线上的有n个顶点;
(1)D x 破坏顶点x;
(2)Q x 表示查询以x所在的最长的连续的点的个数，包括顶点x本身;
(3)R   修复上一次被破坏的点;
算法思想：
线段树，求最大连续区间，操作类似于pku3667;
ls 记录区间左端点开始的最大连续个数，
rs 记录区间右端点开始的最大的连续个数,
ms 表示该区间最大的连续点的个数。
***************************************/
#include<iostream>
#include<algorithm>
#include<stack>
#include<cstring>
#include<cstdlib>
#include<cstdio>
using namespace std;

#define L l,m,u<<1
#define R m+1,r,u<<1|1

const int N=50050;
int st[N];//模拟栈序列
int top;//模拟栈顶元素

struct Node
{
    int l,r;
    int ls,rs,ms;//ls:左边开始连续的最大长度;rs:右边开始最大的连续长度;ms:这个区间最大连续长度;
} T[N*3];

void build(int l,int r,int u)
{
    T[u].l=l;
    T[u].r=r;
    T[u].ls=T[u].rs=T[u].ms=r-l+1;
    if(l==r)
        return;
    int m=(l+r)>>1;
    build(L);
    build(R);
}

void update(int u,int t,int x)
{
    if(T[u].l==T[u].r)
    {
        if(x==1)
            T[u].ls=T[u].rs=T[u].ms=1;
        else
            T[u].ls=T[u].rs=T[u].ms=0;
        return;
    }
    int m=(T[u].l+T[u].r)>>1;
    if(t<=m)
        update(u<<1,t,x);
    else
        update(u<<1|1,t,x);

    T[u].ls=T[u<<1].ls;
    T[u].rs=T[u<<1|1].rs;
    T[u].ms=max(T[u<<1].ms,T[u<<1|1].ms);
    T[u].ms=max(T[u].ms,T[u<<1].rs+T[u<<1|1].ls);

    if(T[u<<1].ls==T[u<<1].r-T[u<<1].l+1)
        T[u].ls+=T[u<<1|1].ls;
    if(T[u<<1|1].rs==T[u<<1|1].r-T[u<<1|1].l+1)
        T[u].rs+=T[u<<1].rs;
}

int query(int u,int t)
{
    if(T[u].l==T[u].r||T[u].ms==0||T[u].ms==T[u].r-T[u].l+1)
    {
        return T[u].ms;
    }
    int m=(T[u].l+T[u].r)>>1;
    if(t<=m)
    {
        if(t>=T[u<<1].r-T[u<<1].rs+1)
            return query(u<<1,t)+query(u<<1|1,m+1);
        else
            return query(u<<1,t);
    }
    else
    {
        if(t<=T[u<<1|1].l+T[u<<1|1].ls-1)
            return query(u<<1|1,t)+query(u<<1,m);
        else
            return query(u<<1|1,t);
    }
}

int main()
{
    //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin);
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        build(1,n,1);
        top=0;
        while(m--)
        {
            char c;
            int x;
            scanf(" %c",&c);
            if(c=='D')
            {
                scanf("%d",&x);
                st[top++]=x;
                update(1,x,0);
            }
            else if(c=='Q')
            {
                scanf("%d",&x);
                printf("%d\n",query(1,x));
            }
            else
            {
                if(x>0)
                {
                    x=st[--top];
                    update(1,x,1);
                }
            }
        }
    }
    return 0;
}