HDU 1540 Tunnel Warfare（平衡树） 解题报告

一道看似线段树的题，用平衡树也破了。。。。

/*
 * 题意：有N个村庄，相邻点相连，可以毁坏某个点或修复毁坏点，求某点的最大联通点数。
 * 方法：平衡树。 
 * 分析：最最朴素的想法就是把破坏的村庄做一个标记，然后遇到Q的时候，就循环的找一
 * 下答案，当然这么做肯定是会超时的设想一下，如果能够很快的找到距离某个点最近的
 * 两个被破坏掉的点，那么答案就很容易得到了于是用到了平衡树，如果一个点被破坏，
 * 则将这个点插入到平衡树中，如果询问，则找到距离X最近的两个在平衡树中的点（一个
 * 大于X，一个小于X），如果是修复，则将这个点从平衡树中删去。
 */
#include<cstdio>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
set<int> ma;
set<int>::iterator it;
//第一次用平衡树，发现真的叼爆了。。。 
int main()
{
    int n,m,i,x;
    char c; 
    while(scanf("%d%d",&n,&m)!=EOF){
        stack<int> s;
        ma.clear();
        ma.insert(0);
        ma.insert(n+1);
        //记得清空与维护边界； 
        while(m--){
            getchar();
            scanf("%c",&c);
            if(c!='R')scanf("%d",&x);
            if(c=='D'){
                ma.insert(x);
                s.push(x);
            }
            else if(c=='R'){
                x=s.top();
                s.pop();
                it = ma.lower_bound(x);
                ma.erase(it);
            }
            else if(c=='Q'){
                 int findl,findr;
                 it = ma.lower_bound(x);
                 if (it != ma.begin() && *it > x) it--;
                 //寻找x在平衡树中的下界，即第一个 <=x 的值的位置； 
                 if (*it == x ) {
                     puts("0");
                     continue;
                 }//如果 x 已被毁坏直接输出； 
                 findl=*it;
                 it++;
                 findr=*it;
                 //printf("%d %d\n",findl,findr);
                 printf("%d\n",findr-findl-1); 
            }
        }
    }      
    return 0;
}