hdu 1540 Tunnel Warfare

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1540

题目大意：给你一排连续的点，你可以进行下面三种操作：

1、删除一个点

2、询问和一个点所在段的长度（如果该点已删除，长度为0）

3、恢复上一个删除的点

线段树计算连续区间，结点保存该段左连续长度lm，和右连续长度rm

/*
 * Problem: hdu 1540 Tunnel Warfare
 * Author:  SHJWUDP
 * Created Time:  2015/3/28 星期六 13:10:49
 * File Name: 233.cpp
 * State: Accepted
 * Memo: 线段树
 */
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MaxA=5e4+7;

struct SegmentTree {
//这个结构可以求最大连续区间，只需要加上一个变量保存当前节点下最大连续区间
    struct Node {    
        int lm, rm;
        Node(){}
        Node(int l, int r):lm(l), rm(r){}
    } *c;
    int n;
    int p, sign;

    SegmentTree(int n):n(n) {
        c=new Node[(n+7)<<2];
        build(1, n, 1);
    }
    ~SegmentTree() {
        delete c;
    }
#define INIT Node& u=c[rt]; Node& ls=c[rt<<1]; Node& rs=c[rt<<1|1];
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1

    void pushUp(int l, int m, int r, int rt) {
        INIT;
        u=Node(ls.lm, rs.rm);
        if(m-l+1==ls.lm) u.lm+=rs.lm;
        if(r-m==rs.rm) u.rm+=ls.rm;
    }

    void build(int l, int r, int rt) {
        if(l==r) {
            c[rt]=Node(1, 1);
        } else {
            int m=(l+r)>>1;
            build(lson);
            build(rson);
            pushUp(l, m, r, rt);
        }
    }

    void doUpdate(int l, int r, int rt) {
        if(l==r) {
            c[rt]=sign?Node(1, 1):Node(0, 0);
        } else {
            int m=(l+r)>>1;
            if(p<=m) doUpdate(lson);
            else doUpdate(rson);
            pushUp(l, m, r, rt);
        }
    }

    void update(int p, int sign) {
        this->p=p; this->sign=sign;
        doUpdate(1, n, 1);
    }

    int doQuery(int l, int r, int rt) {
        INIT;
        if(p-l<u.lm || r-p<u.rm) {
            return p-l<u.lm?u.lm:u.rm;
        } else if(l==r) return 0;
        else {
            int m=(l+r)>>1;
            int res=0;
            if(p<=m) res+=doQuery(lson)+(m-p+1<=ls.rm?rs.lm:0);
            else res+=doQuery(rson)+(p-m<=rs.lm?ls.rm:0);
            return res;
        }
    }

    int query(int p) {
        this->p=p;
        return doQuery(1, n, 1);
    }
#undef INIT
#undef lson
#undef rson
};

int n, m;
int stk[MaxA], top;
int main() {
#ifndef ONLINE_JUDGE
    freopen("in", "r", stdin);
    //freopen("out", "w", stdout);
#endif
    while(~scanf("%d%d", &n, &m)) {
        SegmentTree st(n);
        top=0;
        while(m--) {
            char op[2];
            int x;
            scanf("%s", op);
            switch(op[0]) {
            case 'D': 
                scanf("%d", &x);
                st.update(stk[++top]=x, 0);
                break;
            case 'Q': 
                scanf("%d", &x);
                printf("%d
", st.query(x));
                break;
            case 'R': 
                st.update(stk[top--], 1);
                break;
            }
        }
    }
    return 0;
}