HDU - 1540 Tunnel Warfare (线段树-区间合并)

题目链接：

$\quad$HDU - 1540

题目大意：

$\quad$一个$n$，表示当前有一段连续区间$[1, n]。m$个操作，操作有三种：
$\quad \quad“D \quad x”$，表示将$x$位置的点去掉。
$\quad \quad “R”$，表示将最后去掉的点恢复原状。
$\quad \quad “Q \quad x"$，询问$x$所在的连续区间的长度。

数据范围：

$\quad n, m \leq 50000$

解题思路：

$\quad$线段树区间合并的板子题，代码注释写得很清楚。详见代码。

外话（做题时的小错误）：

$\quad$挂了之后以为是自己线段树哪儿写挫了，查了半天，是主函数里面写错了。注意代码109、110行（原来110行在109行的位置，即后压进栈里，WA），我主要是想稍微优化一点，已经破坏的点就不要再次破坏了，没想到竟然出错了！还是数据来得明了：
$\quad$假如按顺序破坏4, 3, 2, 2，按我之前的后压栈，stack里会有4, 3, 2；如果我现在有连着的四个’R’，那么，本应该第二次恢复2的，结果恢复到了3，这样就导致了错误的结果！所以要先压栈。

//线段树-区间合并
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <queue>
using namespace std;
typedef long long LL;
const int inf = 1 << 30;
const LL INF = 1LL << 60;
const int MaxN = 50010;

int n, q;
stack <int> s;
bool vis[MaxN + 5];
struct segtree
{
    int l, r;
    int ls; //表示 从当前节点所管辖的区间的左端点 往右数的最长连续区间的长度
    int rs; //这个就是从右端点往左数的最长连续区间的长度
    int ms; //表示当前节点所管辖的区间的 最长连续区间的长度
}tree[4 * MaxN + 5];

void Build(int rt, int l, int r)
{
    tree[rt].l = l, tree[rt].r = r;
    tree[rt].ls = r - l + 1; //初始都是所管辖区间的长度
    tree[rt].rs = r - l + 1;
    tree[rt].ms = r - l + 1;
    if(l == r) return ;
    int mid = (l + r) >> 1;
    Build(rt << 1, l, mid);
    Build(rt << 1 | 1, mid + 1, r);
}

void push_up(int rt)
{
    //若左子树全连续,其父亲的 左连续段就为 左子树长度 + 右子树的左连续段
    if(tree[rt << 1].r - tree[rt << 1].l + 1 == tree[rt << 1].ms)
        tree[rt].ls = tree[rt << 1].ms + tree[rt << 1 | 1].ls;
    //否则就直接为左子树的 左连续段
    else tree[rt].ls = tree[rt << 1].ls;

    //右子树更新父亲同理
    if(tree[rt << 1 | 1].r - tree[rt << 1 | 1].l + 1 == tree[rt << 1 | 1].ms)
        tree[rt].rs = tree[rt << 1 | 1].ms + tree[rt << 1].rs;
    else tree[rt].rs = tree[rt << 1 | 1].rs;

    //最后更新父亲的 最长连续段
    tree[rt].ms = max(max(tree[rt << 1].ms, tree[rt << 1 | 1].ms), tree[rt << 1].rs + tree[rt << 1 | 1].ls);
}

void update(int rt, int pos, bool f)
{
    if(tree[rt].r < pos || pos < tree[rt].l) return;
    if(tree[rt].l == tree[rt].r) {
        //1表示被 破坏了
        tree[rt].ls = tree[rt].rs = tree[rt].ms = (f ^ 1);
        return;
    }
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    if(pos <= mid) update(rt << 1, pos, f);
    else update(rt << 1 | 1, pos, f);
    push_up(rt); //向上更新
}

int query(int rt, int pos)
{
    //若当前整个区间都连续 或 整个区间都不连续 或 到达叶子节点,直接返回节点信息
    if(tree[rt].ms == tree[rt].r - tree[rt].l + 1 || tree[rt].ms == 0 || tree[rt].l == tree[rt].r)
        return tree[rt].ms;
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    if(pos <= mid) { //若查询位置在左子树中
        //若在左子树的 右连续段 中,两种方式,上面一种要快些
        if(pos >= tree[rt << 1].r - tree[rt << 1].rs + 1) {
            //直接返回 左子树的右连续段 + 右子树的左连续段,想想
            return tree[rt << 1].rs + tree[rt << 1 | 1].ls;
            /*继续查询 pos 和 右子树左端点,返回它们连续区间的和
            return query(rt << 1, pos) + query(rt << 1 | 1, mid + 1);*/
        }
        //若不在,继续查询左子树
        else return query(rt << 1, pos);
    }
    else { //若在右子树中,与左子树类似
        if(pos <= tree[rt << 1 | 1].l + tree[rt << 1 | 1].ls - 1) {
            return tree[rt << 1 | 1].ls + tree[rt << 1].rs;
            //return query(rt << 1 | 1, pos) + query(rt << 1, mid);
        }
        else return query(rt << 1 | 1, pos);
    }
}

int main()
{
    while(scanf("%d %d", &n, &q) != EOF)
    {
        Build(1, 1, n);
        for(int i = 1; i <= q; i++) {
            char op;
            int x;
            //vis[x] == 1则表示x位置被破坏,否则就没有
            scanf(" %c", &op);
            if(op == 'D') {
                scanf("%d", &x);
                s.push(x); //一定要先压进栈里!
                if(vis[x] == 1) continue;
                vis[x] = 1;
                update(1, x, 1);
            }
            else if(op == 'R') {
                if(s.empty()) continue;
                while(vis[s.top()] == 0) s.pop();
                x = s.top();
                s.pop();
                vis[x] = 0;
                update(1, x, 0);
            }
            else if(op == 'Q') {
                scanf("%d", &x);
                if(vis[x] == 1) printf("0\n");
                else printf("%d\n", query(1, x));
            }
        }
        while(!s.empty()) s.pop();
        memset(vis, 0, sizeof(vis));
        memset(tree, 0, sizeof(tree));
    }
    return 0;
}