Tunnel Warfare poj 2892 hdu 1540（线段树）

做kuangbin专题的时候碰见的，hdu各种超时 wa，然后我就怀疑人生了，感觉我写的也没啥毛病啊，就去翻了discuss，发现。。

大家都被这一题坑了，不过poj上的没问题。

 

解题思路：有点像区间最大连续和，对于每个节点维护一个从左开始的最大连续区间，从右开始的最大连续区间。然后对一点的查询就变成了对于该点前后两个区间的查询。

#include<iostream>
#include<cstdio>
#include<stack>
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define inf 0x3f3f3f3f
#define LL long long
#define fir first
#define sec second
#define lson rt<<1
#define rson rt<<1|1
#define sq(x) (x)*(x)
#define sca(x) scanf("%d",&x)
#define pb(x) push_back(x)
using namespace std;
#define N 50005
#define maxn 2000005

stack<int>S;
struct node
{
    int pre,shr;
}T[N<<2];

void push_up(int rt,int len)
{
    int llen=len-len/2;
    int rlen=len/2;
    if(T[rt<<1].pre==llen) T[rt].pre=T[rt<<1].pre+T[rt<<1|1].pre;
    else T[rt].pre=T[rt<<1].pre;
    if(T[rt<<1|1].shr==rlen) T[rt].shr=T[rt<<1|1].shr+T[rt<<1].shr;
    else T[rt].shr=T[rt<<1|1].shr;
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        T[rt].pre=T[rt].shr=1;
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    push_up(rt,r-l+1);
}

void update(int l,int r,int rt,int idx,int val)
{
    if(l==r)
    {
        T[rt].pre=T[rt].shr=val;
        return ;
    }
    int m=(l+r)>>1;
    if(idx<=m) update(l,m,rt<<1,idx,val);
    if(idx>m ) update(m+1,r,rt<<1|1,idx,val);
    push_up(rt,r-l+1);
}

node query(int ql,int qr,int l,int r,int rt)
{
    if(l>=ql && r<=qr)
    {
        return T[rt];
    }
    int m=(l+r)>>1;
    if(qr<=m) return query(ql,qr,l,m,rt<<1);
    if(ql>m) return  query(ql,qr,m+1,r,rt<<1|1);
    else
    {
        node tmp1=query(ql,m,l,m,rt<<1);
        node tmp2=query(m+1,qr,m+1,r,rt<<1|1);
        node tmp;
        if(tmp1.pre==m-ql+1) tmp.pre=tmp1.pre+tmp2.pre;
        else tmp.pre=tmp1.pre;
        if(tmp2.shr==qr-m) tmp.shr=tmp1.shr+tmp2.shr;
        else tmp.shr=tmp2.shr;
        return tmp;
    }
}

int n,m;
void solve()
{
    char s[3];
    int x;
    scanf("%s%d",s,&x);
    if(s[0]=='D')
    {
        update(1,n,1,x,0);
        S.push(x);
    }
    if(s[0]=='Q')
    {
        node tmp1=query(1,x,1,n,1);
        node tmp2=query(x,n,1,n,1);
        printf("%d\n",tmp1.shr+tmp2.pre>0?tmp1.shr+tmp2.pre-1:0);
    }
    if(s[0]=='R')
    {
        int now=S.top();
        S.pop();
        update(1,n,1,now,1);
    }
}

void init()
{
    while(!S.empty())S.pop();
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF){
    build(1,n,1);
    init();
    while(m--)solve();
    }
    return 0;
}