HDU1540 - Tunnel Warfare(单点修改&&区间合并)

题目大意

给定一个长度为n的全部为1的序列，可以对其进行以下三种操作：

1、D x 把第x个元素修改为0
2、Q x 查询包含第x个元素的全部为1的最长连续序列长度
3、R: 把最后一次修改的元素还原为1

题解

比较水的区间合并问题~~O(∩_∩)O~~。维护两个域即可，即包含节点左端的全部为1的最长连续序列长度lsum，包含节点右端的全部为1的最长连续序列长度rsum，对于某个查询，假设m为节点的中间位置，如果m-rsum[s<<1]+1<=x&&x<=m+lsum[s<<1|1](s为节点编号），则rsum[s<<1]+lsum[s<<1|1]就是查询的结果

代码：

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

using namespace std;

#define MAXN 50005

#define lson l,m,s<<1

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

int lsum[MAXN<<2],rsum[MAXN<<2],a[MAXN];

void PushUp(int s,int m)

{

        lsum[s]=lsum[s<<1];

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

        if(lsum[s<<1]==(m-(m>>1))) lsum[s]+=lsum[s<<1|1];

        if(rsum[s<<1|1]==(m>>1)) rsum[s]+=rsum[s<<1];

}

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

{

        lsum[s]=rsum[s]=r-l+1;

        if(l==r)

                return;

        int m=(l+r)>>1;

        build(lson);

        build(rson);

}

void update(int p,int d,int l,int r,int s)

{

        if(l==r) {

                lsum[s]=rsum[s]=d;

                return;

        }

        int m=(l+r)>>1;

        if(p<=m) update(p,d,lson);

        else

                update(p,d,rson);

        PushUp(s,r-l+1);

}

int query(int p,int l,int r,int s)

{

        if(l==r) return lsum[s];

        int m=(l+r)>>1;

        if(m-rsum[s<<1]+1<=p&&p<=m+lsum[s<<1|1]) return rsum[s<<1]+lsum[s<<1|1];

        if(p<=m) return query(p,lson);

        else

                return query(p,rson);

}

int main(void)

{

        int n,m,p,len;

        char op[5];

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

                len=0;

                build(1,n,1);

                while(m--) {

                        scanf("%s",op);

                        if(op[0]=='D') {

                                scanf("%d",&p);

                                update(p,0,1,n,1);

                                a[len++]=p;

                        } else if(op[0]=='R') {

                                update(a[len-1],1,1,n,1);

                                len--;

                        } else {

                                scanf("%d",&p);

                                printf("%d\n",query(p,1,n,1));

                        }

                }

        }

        return 0;

}