POJ2892 Tunnel Warfare （平衡树）

传送门
思路：当炸掉一个城市的时候，就把这个城市放入平衡树中（最开始的时候放入n+1和0），查询的时候输出这个数的后继-前驱-1，这就是答案。

代码（写的Splay）：

#include<cstdio>
#define MAXN 50005
struct node {
    int v, ch[2], f;
}t[MAXN];
int rt, sz, n, m;
void rot(int x)
{
    int y = t[x].f, z = t[y].f;
    bool f = (t[y].ch[1] == x);
    t[y].ch[f] = t[x].ch[f^1];
    if(t[y].ch[f]) t[t[y].ch[f]].f = y;
    t[x].ch[f^1] = y; t[y].f = x;
    t[x].f = z;
    if(z) t[z].ch[t[z].ch[1]==y] = x;
}
void Spaly(int r, int tp) {
    for(int y, z; (y = t[r].f) != tp; rot(r)) {
        z = t[y].f;
        if(z == tp) continue;
        if( (t[z].ch[0] == y) == (t[y].ch[0] == r) ) rot(y);
        else rot(r);
    }
    if(!tp) rt = r;
}
void Ins(int r, int x) {
    int y = 0;
    while(r && t[r].v != x) { y = r; r = t[r].ch[x > t[r].v]; }
    r = ++ sz;
    t[r].v = x; t[r].f = y; if(y) t[y].ch[x > t[y].v] = r;
    Spaly(r, 0);
}
void Find(int v) {
    int x = rt; if(!x) return;
    while(t[x].ch[v > t[x].v] && t[x].v != v) x = t[x].ch[v > t[x].v];
    Spaly(x, 0);
}
int Nxt(int x, bool f)
{
    Find(x);
    if((t[rt].v>x&&f)||(t[rt].v<x&&!f)) return rt;
    int p = t[rt].ch[f];
    while(t[p].ch[f^1]) p = t[p].ch[!f];
    return p;
}
void Del(int v) {
    int p = Nxt(v, 0), s = Nxt(v, 1);
    Spaly(p, 0); Spaly(s, p);
    p = t[s].ch[0];
    t[s].ch[0] = 0;
}
char c, f;
inline void GET(int &n) {
    n = 0; f = 1;
    do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0');
    while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();}
    n *= f;
}
int op[MAXN], tp;   //用栈来保存炸过的城市
bool dsd[MAXN];     //表示是否被炸过
int main() {
    GET(n); GET(m);
    char s[3]; int x;
    Ins(rt, 0); Ins(rt, n+1);
    while(m --) {
        scanf("%s", s);
        if(s[0] == 'D') GET(op[++ tp]), Ins(rt, op[tp]), dsd[op[tp]] = 1;
        else if(s[0] == 'Q') {
            GET(x); if(dsd[x]) puts("0");
            else printf("%d\n", t[Nxt(x,1)].v-t[Nxt(x,0)].v-1);
        }
        else dsd[op[tp]] = 0, Del(op[tp --]);
    }
    return 0;
}