POJ 2892 Tunnel Warfare（线段树区间合并与查询）

题意：

n个村庄，开始用地下通道连接，可以对其进行 D 摧毁，R 复原最近炸坏的通道，Q 查询 x 村庄所连接的通道数。

思路：

线段树区间查询，没什么特别的。同 HDOJ 1540 http://acm.hdu.edu.cn/showproblem.php?pid=1540 （注意：多个case）

#include <iostream>
#include <stack>
using namespace std;

#define lhs l, m, rt << 1
#define rhs m + 1, r, rt << 1 | 1

const int maxn = 50010;
int sum[maxn << 2], lsum[maxn << 2], rsum[maxn << 2];

void PushUp(int rt, int m)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
    lsum[rt] = lsum[rt << 1];
    rsum[rt] = rsum[rt << 1 | 1];

    if (lsum[rt] == m - (m >> 1))
        lsum[rt] += lsum[rt << 1 | 1];
    if (rsum[rt] == m >> 1)
        rsum[rt] += rsum[rt << 1];
}

void Build(int l, int r, int rt)
{
    sum[rt] = lsum[rt] = rsum[rt] = r - l + 1;

    if (l == r)
        return ;

    int m = (l + r) >> 1;
    Build(lhs);
    Build(rhs);
}

void Update(int p, int value, int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] = lsum[rt] = rsum[rt] = value;
        return ;
    }

    int m = (l + r) >> 1;
    if (p <= m)
        Update(p, value, lhs);
    else
        Update(p, value, rhs);
    PushUp(rt, r - l + 1);
}

int Query(int p, int l, int r, int rt)
{
    if (l == r)
        return sum[rt];

    int m = (l + r) >> 1;
    if (p <= m)
    {
        if (p + rsum[rt << 1] > m)
            return rsum[rt << 1] + lsum[rt << 1 | 1];
        else
            return Query(p, lhs);
    }
    else
    {
        if (m + lsum[rt << 1 | 1] >= p)
            return rsum[rt << 1] + lsum[rt << 1 | 1];
        else
            return Query(p, rhs);
    }
}

int main()
{
    int n, m;
    stack<int> num;

    scanf("%d %d", &n, &m);
    Build(1, n, 1);

    for (int i = 0; i < m; ++i)
    {
        char op[12];
        int x;
        scanf("%s", op);

        if (op[0] == 'D')
        {
            scanf("%d", &x);
            num.push(x);
            Update(x, 0, 1, n, 1);
        }
        else if (op[0] == 'R')
        {
            if (!num.empty())
            {
                Update(num.top(), 1, 1, n, 1);
                num.pop();
            }
        }
        else if (op[0] == 'Q')
        {
            scanf("%d", &x);
            printf("%d\n", Query(x, 1, n, 1));
        }
    }
    return 0;
}