hdu1540

怎么会T啊

/*
三种操作：D x：第x个位置1
         Q x：查询第x位置所在的0连续块
         R ：将上次D的位置置0
*/
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 50050
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int lmx[maxn<<],rmx[maxn<<],mx[maxn<<];
int des[maxn],top;

inline void pushup(int l,int r,int rt){
    lmx[rt]=lmx[rt<<];
    rmx[rt]=rmx[rt<<|];
    mx[rt]=max(mx[rt<<],mx[rt<<|]);

    int m=l+r>>;
    if(lmx[rt<<]==m-l+) lmx[rt]=m-l++lmx[rt<<|];
    if(rmx[rt<<|]==r-l) rmx[rt]=r-m+rmx[rt<<];
    mx[rt]=max(mx[rt],rmx[rt<<]+lmx[rt<<|]);
}
void build(int l,int r,int rt){
    if(l==r){
        lmx[rt]=rmx[rt]=mx[rt]=;
        return;
    }
    int m=l+r>>;
    build(lson);
    build(rson);
    pushup(l,r,rt);
}
void update(int pos,int c,int l,int r,int rt){
    if(l==r){
         lmx[rt]=rmx[rt]=mx[rt]=c;
        return;
    }
    int m=l+r>>;
    if(pos<=m) update(pos,c,lson);
    else if(pos>m) update(pos,c,rson);
    pushup(l,r,rt);
}
int query(int pos,int l,int r,int rt){
    if(mx[rt]== || l==r || mx[rt]==r-l+)
        return mx[rt];

    int m=l+r>>;
    if(pos<=m){
        if(pos>=m-rmx[rt<<]+)
            return query(pos,lson)+query(m+,rson);
        else return query(pos,lson);
    }
    else {
        if(pos<=m+lmx[rt<<|])
            return query(pos,rson)+query(m,lson);
        else return query(pos,rson);
    }
}
int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(des,,sizeof des);
        top=;
        build(,n,);

        char op[];
        int a,b;
        while(m--){
            scanf("%s",op);
            if(op[]=='D'){
                scanf("%d",&a);
                update(a,,,n,);
                des[top++]=a;
            }
            else if(op[]=='Q'){
                scanf("%d",&a);
                printf("%d\n",query(a,,n,));
            }
            else {
                int tmp=des[--top];
                update(tmp,,,n,);
            }
        }
    }
    return ;
}

ac代码

/*
三种操作：D x：第x个位置1
         Q x：查询第x位置所在的0连续块
         R ：将上次D的位置置0
*/
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 50050
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int lmx[maxn<<],rmx[maxn<<],mx[maxn<<];
int des[maxn],top;

inline void pushup(int l,int r,int rt){
    lmx[rt]=lmx[rt<<];
    rmx[rt]=rmx[rt<<|];
    mx[rt]=max(mx[rt<<],mx[rt<<|]);

    int m=l+r>>;
    if(lmx[rt<<]==m-l+) lmx[rt]=m-l++lmx[rt<<|];
    if(rmx[rt<<|]==r-m) rmx[rt]=r-m+rmx[rt<<];
    mx[rt]=max(mx[rt],rmx[rt<<]+lmx[rt<<|]);
}
void build(int l,int r,int rt){
    if(l==r){
        lmx[rt]=rmx[rt]=mx[rt]=;
        return;
    }
    int m=l+r>>;
    build(lson);
    build(rson);
    pushup(l,r,rt);
}
void update(int pos,int c,int l,int r,int rt){
    if(l==r){
         lmx[rt]=rmx[rt]=mx[rt]=c;
        return;
    }
    int m=l+r>>;
    if(pos<=m) update(pos,c,lson);
    if(pos>m) update(pos,c,rson);
    pushup(l,r,rt);
}
int query(int pos,int l,int r,int rt){
    if(mx[rt]== || l==r || mx[rt]==r-l+)
        return mx[rt];

    int m=l+r>>;
    if(pos<=m){
        if(pos+rmx[rt<<]>m)//比之前优化了
            return rmx[rt<<]+lmx[rt<<|];
        else return query(pos,lson);
    }
    else {
        if(m+lmx[rt<<|]>=pos)//比之前优化了
            return rmx[rt<<]+lmx[rt<<|];
        else return query(pos,rson);
    }
}
int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(des,,sizeof des);
        top=;
        build(,n,);

        char op[];
        int a,b;
        while(m--){
            scanf("%s",op);
            if(op[]=='D'){
                scanf("%d",&a);
                update(a,,,n,);
                des[top++]=a;
            }
            else if(op[]=='Q'){
                scanf("%d",&a);
                printf("%d\n",query(a,,n,));
            }
            else {
                int tmp=des[--top];
                update(tmp,,,n,);
            }
        }
    }
    return ;
}