hdu 1540 区间合并

话说就这样一道很水的区间合并，搞了差不多20次WA。。。昨天搞得相当头疼，今天中午终于搞掂了。。。嘿嘿。。

题目是说的比较清楚，实际上就是查询与某村庄连续的村庄有多少，设所有村庄初值为0，若这个村庄被日本鬼子毁掉，则变为1，纪录区间从右到左有多少个0或从左到右有多少个0即可。。pushup函数和更新都比较简单，关键在于查询。。。以查询某区间最右边有连续多少个0为例（右查询），先要查询其右子区间，我刚开始认为当右子区间全为0时，再查询左子区间。。这只是一种情况，还有两种。。。下面说我没神马WA这么多次的原因，一个是查询处理没有做好，更重要的是返回结果一直是爆栈，这个刚开始让我完全想不通，后来自己想了一组测试数据，才懂。。。哎。。原来是当Q后面为1时，不应作左查询，这样就出现了0的情况。。。。附代码:

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <vector>

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ss(a) scanf("%d",&a)
#define cl(a) memset(a,0,sizeof(a))
#define N 50005
#define pub push_back
#define pob pop_back

using namespace std;

int sum[3][N<<3],a[N];
vector<int>b;

void pushup(int l,int r,int rt)
{
    int m=(l+r)>>1;
    sum[0][rt]=sum[0][rt<<1];
    if (sum[0][rt<<1]==m-l+1) sum[0][rt]+=sum[0][rt<<1|1];
    sum[1][rt]=sum[1][rt<<1|1];
    if (sum[1][rt<<1|1]==r-m) sum[1][rt]+=sum[1][rt<<1];
}

void build(int l,int r,int rt)
{
    if (l==r)
    {
        sum[0][rt]=1;
        sum[1][rt]=1;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(l,r,rt);
}

void update(int x,int c,int l,int r,int rt)
{
    if (l==r)
    {
        sum[0][rt]+=c;
        sum[1][rt]+=c;
        return;
    }
    int m=(l+r)>>1;
    if (x<=m) update(x,c,lson);
    else update(x,c,rson);
    pushup(l,r,rt); 
}

int lquery(int L,int R,int x,int l,int r,int rt)
{
    if (L<=l&&r<=R)
    {
        return sum[0][rt];
    }
    int m=(l+r)>>1,res=0;
    if (L<=m) res+=lquery(L,R,x,lson);
    if (R>m) 
    {
        int flag=0;
        if (L>m) flag=1;   //查询区间端点在右子区间时，不存在左子区间值 
        if (res==m-l+1||res==m-x) flag=1;            //左子区间全是0或者在查询点之后全为0 
        if (flag) res+=lquery(L,R,x,rson);
    }
    return res; 
}

int rquery(int L,int R,int x,int l,int r,int rt)
{
    if (L<=l&&r<=R)
    {
        return sum[1][rt];
    }
    int m=(l+r)>>1,res=0;
    if (R>m) res+=rquery(L,R,x,rson);
    if (L<=m)
    {
        int flag=0;
        if (R<=m) flag=1;
        if (res==r-m||res==x-1-m) flag=1;
        if (flag) res+=rquery(L,R,x,lson);
    }
    return res; 
} 

int main()
{
    int n,i,m,x;
    while (ss(n)!=EOF)
    {
        ss(m);
        cl(sum);cl(a);
        b.clear();
        build(1,n,1);
        for (i=1;i<=m;i++)
        {
            char ss[3];
            scanf("%s",ss);
            if (ss[0]=='R')
            {
                if (!b.empty())
                {
                    int k=b.back();
                    a[k]=0;
                    update(k,1,1,n,1);
                    b.pob();    
                }
            }
            else if (ss[0]=='D')
            {
                ss(x);
                b.pub(x);
                if (!a[x]) 
                {
                    a[x]=1;
                    update(x,-1,1,n,1);
                }
            }
            else
            {
                ss(x);
                int res=0;
                if (a[x]==0) 
                {
                    res=1;
                    if (x>1) res+=rquery(1,x-1,x,1,n,1);  //就是这个条件没加，搞得一直爆栈。。_ _ 
                    if (x<n) res+=lquery(x+1,n,x,1,n,1);
                }
                printf("%d\n",res);
            }
        }
    }
    return 0;
}